def filter_by_extension(self, extensions): """Keep only files with given extensions (e.g. ['.nes', '.sfc']).""" extensions = [ext.lower() if ext.startswith('.') else f'.ext.lower()' for ext in extensions] self.roms = [r for r in self.roms if r.suffix.lower() in extensions] return self
def filter_by_name_pattern(self, pattern, case_sensitive=False): """Keep ROMs whose name matches regex pattern.""" flags = 0 if case_sensitive else re.IGNORECASE regex = re.compile(pattern, flags) self.roms = [r for r in self.roms if regex.search(r.name)] return self romlister
if args.extensions: lister.filter_by_extension(args.extensions) if args.min_size is not None or args.max_size is not None: lister.filter_by_size(args.min_size, args.max_size) if args.pattern: lister.filter_by_name_pattern(args.pattern) clean_names=False): with open(filepath
[ "/home/user/roms/nes/Super Mario Bros (USA).nes", "/home/user/roms/nes/Legend of Zelda (USA).nes" ] encoding='utf-8') as f: json.dump(self.get_list(clean_names)
def export_json(self, filepath, clean_names=False): with open(filepath, 'w', encoding='utf-8') as f: json.dump(self.get_list(clean_names), f, indent=2) CLI Interface ------------------------------ def main(): parser = argparse.ArgumentParser(description="RomLister - Scan and filter ROM collections") parser.add_argument("directory", help="Root directory to scan for ROMs") parser.add_argument("-r", "--recursive", action="store_true", help="Scan subfolders recursively") parser.add_argument("-e", "--extensions", nargs="+", help="File extensions to include (e.g., nes sfc iso)") parser.add_argument("--min-size", type=int, help="Minimum file size in bytes") parser.add_argument("--max-size", type=int, help="Maximum file size in bytes") parser.add_argument("-p", "--pattern", help="Regex pattern to filter by filename") parser.add_argument("-c", "--clean-names", action="store_true", help="Remove region/version tags") parser.add_argument("-o", "--output", help="Output file path") parser.add_argument("-f", "--format", choices=["txt", "csv", "json"], default="txt", help="Output format") parser.add_argument("--list-only", action="store_true", help="Print list to console")
args = parser.parse_args()