#import tkinter as tk
import sys
import os
import shutil
from datetime import datetime
import time
from PyQt5 import QtCore
from PyQt5.QtWidgets import *
from PyQt5.QtGui import QIcon, QPixmap
from PyQt5.QtCore import *

scriptPath = os.path.dirname(os.path.realpath(__file__))
filePath = scriptPath + '\\files\\input'

def files(path):
    for file in os.listdir(path):
        if os.path.isfile(os.path.join(path, file)):
            yield file


def save_input(oldpath):
    # make timestampt the filename, so it wouln't overwrite
    timestamp = str(int(time.time()))
    filename = timestamp + '.' + oldpath.split('.')[-1]
    newpath = filePath + '\\' + filename
    shutil.copy(oldpath, newpath)
    return newpath
    

class LibraryTableButtons(QWidget):
    def __init__(self, filename, parent=None):
        super(LibraryTableButtons,self).__init__(parent)

        def viewFile():
            os.startfile(filePath + '\\' + filename)

        def deleteFile():
            os.remove(filePath + '\\' + filename)

        layout = QHBoxLayout()
        layout.setContentsMargins(0,0,0,0)
        layout.setSpacing(0)

        view_btn = QPushButton('View')
        view_btn.clicked.connect(viewFile)
        layout.addWidget(view_btn)

        delete_btn = QPushButton('Delete')
        delete_btn.clicked.connect(deleteFile)
        layout.addWidget(delete_btn)

        self.setLayout(layout)


class LibraryTable(QTableWidget):

    def __init__(self, parent=None):
        QTableWidget.__init__(self)
        self.setColumnCount(3)
        self.setHorizontalHeaderLabels(['Upload date', 'Filename', 'Options'])
        self.horizontalHeader().setSectionResizeMode(0, QHeaderView.ResizeToContents)
        self.horizontalHeader().setSectionResizeMode(1, QHeaderView.Stretch)
        self.horizontalHeader().setSectionResizeMode(2, QHeaderView.ResizeToContents)

        dates = []
        names = []
        for index, file in enumerate(files(filePath)):
            dates.append(QTableWidgetItem(str(datetime.fromtimestamp(os.path.getmtime(filePath + '/' + file)))))
            names.append(QTableWidgetItem(str(file)))

        self.setRowCount(len(dates))

        for index, date in enumerate(dates):
            self.setItem(index,0,date)
            self.setItem(index,1,names[index])
            self.setCellWidget(index,2,LibraryTableButtons(names[index].text()))


class MainWindow(QMainWindow):

    def __init__(self):
        super().__init__()
        self.initUI()
        
    # Show raw uploaded files
    def showLibrary(self):
        libTable = LibraryTable()
        self.setCentralWidget(libTable)
        
    def showUploadFile(self):
        dialog = QFileDialog(self)
        dialog.setFileMode(QFileDialog.AnyFile)
        dialog.setFilter(QDir.Files)
        if dialog.exec_():
            file_path = dialog.selectedFiles()[0] # ['C:/Users/agatha/Desktop/SYI/VisionScore/win_venv/requirements.txt']
            save_input(file_path)
        
    def initUI(self):
        self.setGeometry(0, 0, 600, 400) 
        self.setWindowTitle('VisionScore')
        scriptDir = os.path.dirname(os.path.realpath(__file__))
        self.setWindowIcon(QIcon(scriptDir + os.path.sep + 'static/v_logo.jpg'))
        
        # Toolbar
        menuBar = self.menuBar()
        homeMenu = QMenu("&Home", self)
        menuBar.addMenu(homeMenu)
        
        # Upload file
        uploadAct = QAction('&Upload new file', self)
        uploadAct.triggered.connect(self.showUploadFile)
        homeMenu.addAction(uploadAct)

        libraryAct = QAction('&Library', self)
        libraryAct.triggered.connect(self.showLibrary)
        menuBar.addAction(libraryAct)

        # Exit app
        exitAct = QAction('&Exit', self)
        exitAct.setShortcut('Ctrl+Q')
        exitAct.setStatusTip('Exit')
        exitAct.triggered.connect(qApp.quit)
        homeMenu.addAction(exitAct)
        
        helpMenu = menuBar.addMenu("&Help")  
        
        self.show()


def main():
    app = QApplication(sys.argv)

    w = MainWindow()
    sys.exit(app.exec_())


if __name__ == '__main__':
    main()