import os
from AppCode.Config import DBConfig
from AppCode.FileHandler import FileHandler
from werkzeug.utils import secure_filename

class DocumentHandler:
    
    years = ""
    documents = ""
    isSuccess = False
    resultMessage = ""

    def View(self,request):
        year = request.args.get('year')
        stage = request.args.get('stage')
        dbconfig = DBConfig()
        connection = dbconfig.get_db_connection()

        if not connection:
            self.isSuccess = False
            return
        cursor = connection.cursor()
 
        params = []
        query = "SELECT * FROM documents WHERE 1=1"

        if year:
            query += " AND year = %s"
            params.append(year)
        if stage:
            query += " AND stage = %s"
            params.append(stage)

            
        cursor.execute(query, params)
        documentsdata = cursor.fetchall()
        print("*************")
        print(documentsdata)
        cursor.callproc("GetYear")

        # records = []
        # for result in cursor.stored_results():
        #     records = result.fetchall()

        yearsdata = ""
        for res in cursor.stored_results():
            yearsdata = res.fetchall()
        print(yearsdata)

        self.years = yearsdata
        self.documents = documentsdata
        self.isSuccess = True

        print("document --",documentsdata)

        cursor.close()
        connection.close()


    def Upload(self, request):
        """Log user actions with timestamp, user, action, and details."""
        
        dbconfig = DBConfig()
        connection = dbconfig.get_db_connection()

        if connection:
            cursor = connection.cursor()
            files = request.files.getlist('documents')
            year = request.form['year']
            stage = request.form['stage']
            
            for file in files:
                if file is not FileHandler.ALLOWED_EXTENSIONS:
                    continue


                filename = secure_filename(file.filename)
                filepath = os.path.join(FileHandler.UPLOAD_FOLDER, filename)
                extension = file.filename.rsplit('.', 1)[1]
                # Need to Check whetehr all three items are required

                file.save(filepath)

                # cursor.execute("""
                #     INSERT INTO documents (filename, filepath, filetype, year, stage)
                #     VALUES (%s, %s, %s, %s, %s)
                # """, (filename, filepath, file.filename.rsplit('.', 1)[1], year, stage))


                cursor.callproc('InsertDocument', [
                    filename,
                    filepath,
                    extension,
                    year,
                    stage
                ])

            connection.commit()
            cursor.close()
            connection.close()
        # return redirect(url_for('view_documents'))

    #return render_template('upload.html')