read sauce info. Required changes to sauce library.

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