sql

import mysql.connector
import logs
import shutil
import datetime
mydb = mysql.connector.connect(host = 'localhost',
                               user = 'root',
                               password = 'root',
                               database = 'testdb')

mycursor = mydb.cursor()

def last_id():
    arr = "SELECT * FROM docs WHERE doc_id"
    mycursor.execute(arr)
    myresult=mycursor.fetchall()
    return(myresult[-1][2])

def find_file():
    print("Введите имя файла")
    column_arg=input()
    find="SELECT * FROM docs WHERE name LIKE '%{column_arg}%'".format(column_arg=column_arg) #column_arg - аргумент колонны
    mycursor.execute(find)
    myresult=mycursor.fetchall()
    for result in myresult:
        print(result)

def find_file_id(file_id):
    find = "SELECT * FROM docs WHERE doc_id LIKE '%{column_arg}%'".format(column_arg=file_id)  # column_arg - аргумент колонны
    mycursor.execute(find)
    myresult = mycursor.fetchall()
    return myresult[0][2]

def find_template():
    print("Введите имя шаблона")
    column_arg=input()
    find="SELECT * FROM templates WHERE Name LIKE '%{column_arg}%'".format(column_arg=column_arg) #column_arg - аргумент колонны
    mycursor.execute(find)
    myresult=mycursor.fetchall()
    for result in myresult:
        print(result)

def delete_file(uid, uid_name):
    print("Введите название файла в БД")  # В фронт-версии - заменить
    file_name = input()  # В фронт-версии - заменить
    print("Введите id файла")  # В фронт-версии - заменить
    file_id = int(input())  # В фронт-версии - заменить
    source=r'C:\Users\mi\SAD\Server\docs\\'
    shutil.move(source+file_name+str(file_id), r'C:\Users\mi\SAD\Server\trash\\')
    delete="DELETE FROM docs WHERE doc_id = {file_id}".format(file_id=file_id)
    mycursor.execute(delete)
    logs.log_edit(uid_name, uid, file_name, file_id, text=' удален')
    mydb.commit()

def add_file(uid,uid_name): #file_name, source, doc_id, uid, uid_name
    print("Введите точный путь файла") # В фронт-версии - заменить
    user_source=input() #В фронт-версии - заменить
    print('Введите уровень доступа')
    access_lvl=int(input())
    source=r'C:\Users\mi\SAD\Server\docs\\'
    file_id=last_id()+1
    file_name=user_source.split("\\")[-1]+str(file_id)
    shutil.copy(user_source, source)
    sqlFormula = "INSERT INTO docs (name, source, doc_id, edit_date, access_lvl) VALUES (%s, %s, %s, %s, %s)"
    file=(file_name,source, file_id)
    mycursor.execute(sqlFormula, file)
    logs.log_create(uid_name, uid, file_name, file_id, str(datetime.datetime.now())[:-7], access_lvl)
    mydb.commit()


def edit_file(uid,uid_name):
    print("Введите название файла в БД") # В фронт-версии - заменить
    file_name=input() # В фронт-версии - заменить
    print("Введите id файла") # В фронт-версии - заменить
    file_id=int(input()) # В фронт-версии - заменить
    print("Введите точный путь файла")  # В фронт-версии - заменить
    user_source = input()  # В фронт-версии - заменить
    source = r'C:\Users\mi\SAD\Server\docs\\'
    shutil.copy(user_source, source)
    logs.log_edit(uid_name, uid, file_name, file_id, text=' изменен')
    mydb.commit()

def download_file(uid_access_lvl):
    print("Введите id файла")
    doc_id=int(input())
    print("Введите куда скачать")
    user_source=input()
    file=find_file_id(doc_id)
    doc_access_lvl=file[4]
    if uid_access_lvl>doc_access_lvl:
        shutil.copy(user_source, file[1])

#sqlFormula = "INSERT INTO users (name, uid, access_lvl, hash_pass) VALUES (%s, %s, %s, %s)"
#student1=("Admin", 1, 0, "acbd9ab2f68bea3f5291f825416546a1")
#mycursor.execute(sqlFormula,student1)
#find_file('students', 'name', 'E')

#add_file(r'C:\Users\mi\SAD\User2\test2.txt',1)
# find="SELECT * FROM students WHERE name = 'Rachel' AND age = '19'"
#mycursor.execute(find)
#myresult=mycursor.fetchall()
#print(myresult)
#for result in myresult:
#    print(result)

#mydb.commit()

#mycursor.execute("CREATE TABLE docs (name VARCHAR(255), source VARCHAR(400), doc_id INTEGER(10)) ")

#mycursor.execute("SHOW TABLES")

#test()