#!/usr/bin/python3

from os import listdir, remove
from os.path import isfile, join, getsize
import hashlib, pymysql

files = [f for f in listdir('files') if isfile(join('files', f))]

db = pymysql.connect('localhost',
                     'kingofdog',
                     '123456',
                     'kingofdog')
cur = db.cursor()

def pushToDatabase(fileName, originalName, fileType, size, path, uploadDate, isUserData):
    try:
        cur.execute('INSERT INTO files (name, original_name, type, size, path, isUserData) VALUES (%s, %s, %s, %s, %s, %s)', (fileName, originalName, fileType, size, path, isUserData))
        db.commit()
    except:
        db.rollback()

def checkInDatabase(name):
    cur.execute('SELECT * FROM files WHERE name = %s', (name))
    result = cur.fetchone()
    return result != None
		
def findUnregisteredImages():
    for file in files:
        with open('./files/' + file, 'rb') as imageFile:
            imageName = file.split(".")[0]
            if(not checkInDatabase(imageName)):
                print('Adding "' + imageName + '" to the database.')
                fileType = 'image/jpeg'
                size = getsize(imageFile.name)
                path = imageFile.name[2:]
                pushToDatabase(imageName, file, fileType, size, path, None, False)


print('Searching for unregistered files...')
findUnregisteredImages()
print('Successfully finished registering all previously unregistered files.')

db.close()