144 lines
4.3 KiB
Python
144 lines
4.3 KiB
Python
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
|