44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
|
#!/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()
|