diff --git a/editor.py b/editor.py index 2ddfa05..d7c5a27 100644 --- a/editor.py +++ b/editor.py @@ -9,6 +9,7 @@ import json import argparse import os +from sauce import SAUCE from sahliEditorPython import sahlifile as SF @@ -17,8 +18,6 @@ def getfilesindir(directory): for root, dirs, files, rootfd in os.fwalk(directory): return files -# ---------------------------------------------------------------------- - def getfilenames(filedata): """return the file names from a sahli filedata array""" @@ -36,6 +35,43 @@ def getdata(filedata, name): return [] +def getpicdata(filename): + """extract picture data from filename""" + return 'todo.txt' + + +def getansidata(filename): + """extract SAUCE data from filename""" + saucedata = SAUCE(filename) + ansidata = { + 'author': saucedata.author, + 'group': saucedata.group, + 'title': saucedata.title, + 'filesize': saucedata.filesize, + 'comments': saucedata.comments + } + tinfonames = [saucedata.tinfo1_name, + saucedata.tinfo2_name, + saucedata.tinfo3_name, + saucedata.tinfo4_name] + tinfo = [saucedata.tinfo1, + saucedata.tinfo2, + saucedata.tinfo3, + saucedata.tinfo4] + for i in range(0, 3): + if tinfonames[i] == 'width': + ansidata['width'] = tinfo[i] + if tinfonames[i] == 'height': + ansidata['height'] = tinfo[i] +# print(tinfonames[i]) + return ansidata + + +def getamigadata(filename): + """try to get some form of info from file (:""" + return 'todo.txt' + + def main(args): """maintain a list.sahli file""" if args.new: @@ -48,11 +84,25 @@ def main(args): filedatanames = getfilenames(filedata) newdata = [] for i in files: + dirfile = '{}/{}'.format(args.directory, i) if i in filedatanames: print('found! {}'.format(i)) + a = getansidata(dirfile) newdata.append(getdata(filedata, i)) else: print('not found! {}'.format(i)) + suf = i.split('.')[-1] + if suf in ['png', 'jpg', 'jpeg', 'gif', + 'PNG', 'JPG', 'JPEG', 'GIF']: + stuff = getpicdata(dirfile) + elif suf in ['ans', 'ANS', 'BIN', 'bin', 'XB', 'xb']: + stuff = getansidata(dirfile) + a = 5 + elif suf in ['TXT', 'ASC', 'txt', 'asc', + 'NFO', 'nfo', 'diz', 'DIZ']: + stuff = getamigadata(dirfile) + else: + print("dunno what type of file this is...") a = 5