mirror of https://github.com/evilhero/mylar
58 lines
2.6 KiB
Python
Executable File
58 lines
2.6 KiB
Python
Executable File
import mylar
|
|
from mylar import db, logger, helpers
|
|
import os
|
|
import shutil
|
|
|
|
|
|
def movefiles(comicid, comlocation, ogcname, imported=None):
|
|
myDB = db.DBConnection()
|
|
logger.fdebug('comlocation is : ' + str(comlocation))
|
|
logger.fdebug('original comicname is : ' + str(ogcname))
|
|
impres = myDB.select("SELECT * from importresults WHERE ComicName=?", [ogcname])
|
|
|
|
if impres is not None:
|
|
#print ("preparing to move " + str(len(impres)) + " files into the right directory now.")
|
|
for impr in impres:
|
|
srcimp = impr['ComicLocation']
|
|
orig_filename = impr['ComicFilename']
|
|
orig_iss = impr['impID'].rfind('-')
|
|
orig_iss = impr['impID'][orig_iss +1:]
|
|
logger.fdebug("Issue :" + str(orig_iss))
|
|
#before moving check to see if Rename to Mylar structure is enabled.
|
|
if mylar.IMP_RENAME and mylar.FILE_FORMAT != '':
|
|
logger.fdebug("Renaming files according to configuration details : " + str(mylar.FILE_FORMAT))
|
|
renameit = helpers.rename_param(comicid, impr['ComicName'], orig_iss, orig_filename)
|
|
nfilename = renameit['nfilename']
|
|
dstimp = os.path.join(comlocation, nfilename)
|
|
else:
|
|
logger.fdebug("Renaming files not enabled, keeping original filename(s)")
|
|
dstimp = os.path.join(comlocation, orig_filename)
|
|
|
|
logger.info("moving " + str(srcimp) + " ... to " + str(dstimp))
|
|
try:
|
|
shutil.move(srcimp, dstimp)
|
|
except (OSError, IOError):
|
|
logger.error("Failed to move files - check directories and manually re-run.")
|
|
logger.fdebug("all files moved.")
|
|
#now that it's moved / renamed ... we remove it from importResults or mark as completed.
|
|
|
|
results = myDB.select("SELECT * from importresults WHERE ComicName=?", [ogcname])
|
|
if results is not None:
|
|
for result in results:
|
|
controlValue = {"impID": result['impid']}
|
|
newValue = {"Status": "Imported"}
|
|
myDB.upsert("importresults", newValue, controlValue)
|
|
return
|
|
|
|
def archivefiles(comicid, ogcname):
|
|
myDB = db.DBConnection()
|
|
# if move files isn't enabled, let's set all found comics to Archive status :)
|
|
result = myDB.select("SELECT * FROM importresults WHERE ComicName=?", [ogcname])
|
|
if result is None: pass
|
|
else:
|
|
ogdir = result['Location']
|
|
origdir = os.path.join(os.path.dirname(ogdir))
|
|
|
|
updater.forceRescan(comicid, archive=origdir) #send to rescanner with archive mode turned on
|
|
|