Initial commit

csvfile.py

@@ -0,0 +1,143 @@
+import csv
+import os
+import zipfile
+
+
+class csvROM(object):
+    ''' One CSV ROM '''
+
+    def __init__(self):
+        self.name = ''
+        self.cloneof = None
+        self.description = ''
+        self.year = ''
+        self.manufacturer = ''
+        self.status = ''
+        self.category = ''
+        self.snap = None
+
+    def __str__(self):
+        return self.description
+
+    def __repr__(self):
+        return '%s("%s", "%s", "%s", "%s", "%s", "%s", "%s")' % (
+            self.__class__.__name__, self.name, self.cloneof, self.description,
+            self.year, self.manufacturer, self.status, self.category)
+
+
+class csvROMs(object):
+    ''' A Collection of CSV Roms '''
+
+    def __init__(self, data_dir, cfg):
+        self.all_roms = []
+        self.roms = []
+        self.all_categories = []
+        self.categories = []
+        self.catdict = {}
+        self.len = 0
+        self.data_dir = data_dir
+        self.cfg = cfg
+        self.emulator_dir = os.path.split(cfg['EXE'])[0]
+
+        self.parse()
+        self.filter()
+
+        self.snapdir = cfg['SnapDir']
+        if os.path.isfile(os.path.join(self.snapdir, 'snap.zip')):
+            self.snapfile = zipfile.ZipFile(
+                os.path.join(self.snapdir, "snap.zip"))
+        else:
+            self.snapfile = None
+
+    def __len__(self):
+        return self.len
+
+    def __getitem__(self, item):
+        return self.roms[item]
+
+    def filter(self):
+        self.roms = []
+
+        for rom in self.all_roms:
+            if self.cfg["Category"] == "All Games" or self.cfg["Category"] == rom.category:
+                self.roms.append(rom)
+
+        if self.cfg["Sort"] == "Name Asc":
+            self.roms.sort(key=lambda x: x.description.lower(), reverse=False)
+        elif self.cfg["Sort"] == "Name Dec":
+            self.roms.sort(key=lambda x: x.description.lower(), reverse=True)
+        elif self.cfg["Sort"] == "Year Asc":
+            self.roms.sort(key=lambda x: x.year.lower(), reverse=False)
+        elif self.cfg["Sort"] == "Year Dec":
+            self.roms.sort(key=lambda x: x.year.lower(), reverse=True)
+
+        self.len = len(self.roms)
+
+        self.categories = self.all_categories
+
+    def parse(self):
+        ''' Parse csv file '''
+        if os.path.isfile(self.cfg['CSVFile']):
+            with open(self.cfg['CSVFile']) as f:
+                reader = csv.DictReader(f)
+
+                for game in reader:
+                    rom = csvROM()
+                    rom.name = game['romname']
+                    rom.description = game[
+                        'description'] if 'description' in game.keys(
+                        ) else game['romname']
+                    rom.snap = game['snap'] if 'snap' in game.keys() else game[
+                        'romname']
+                    rom.year = game[
+                        'year'] if 'year' in game.keys() else 'Unknown Year'
+                    rom.category = game[
+                        'category'] if 'category' in game.keys() else 'Unknown'
+                    self.all_roms.append(rom)
+
+            self.all_categories.insert(0, 'All Games')
+            self.all_categories.append('Unknown')
+
+        return True
+
+    def get_snap(self, rom, type_of_image):  # pylint : disable=R0912
+        data = None
+        image_file = None
+        image_dir = None
+
+        if type_of_image == 'snap':
+            image_dir = self.snapdir
+            image_file = self.snapfile
+
+        if image_file:
+            try:
+                data = image_file.read(rom.name + ".png")
+            except:
+                try:
+                    data = image_file.read(rom.cloneof + '.png')
+                except:
+                    data = None
+        elif image_dir:
+            fn = os.path.join(image_dir, rom.name + '.png')
+            if os.path.isfile(fn):
+                with open(fn, 'rb') as f:
+                    data = f.read()
+            elif rom.cloneof:
+                fn = os.path.join(image_dir, rom.cloneof + '.png')
+                if os.path.isfile(fn):
+                    with open(fn, 'rb') as f:
+                        data = f.read()
+                else:
+                    data = None
+            else:
+                data = None
+        else:
+            data = None
+
+        return data
+
+    def get_info(self, rom):
+        return
+
+    def get_history(self, rom):
+        return