AO model code commit
This commit is contained in:
103
test.py
103
test.py
@@ -1,103 +0,0 @@
|
||||
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')
|
||||
Reference in New Issue
Block a user