ldcheck: comparison ldcheck.py

-:eb4c767522ac
+:fde18c4d6784
 #!/usr/bin/env python3
+import argparse
 from sys import version_info
 if version_info < (3, 4):
 raise RuntimeError('Python 3.4 or newer required')
 appname = 'ldcheck'
 version = (1, 0, 9999)
 version_string = str.join('.', map(str, version))
-from colours import load_colours
 from geometry import *
 from pathlib import Path
 import linetypes
 import header
 import parse
 from os.path import realpath
 script_directory = Path(realpath(__file__)).parent
-def load_config(filename = None):
+def config_dirs():
-if filename is None:
+import appdirs
-filename = script_directory / 'ldcheck.cfg'
+appauthor = 'Teemu Piippo'
-from configobj import ConfigObj
+dirs = appdirs.AppDirs(appname, appauthor)
-from copy import deepcopy
+return {
-config = ConfigObj(str(filename), encoding = 'UTF8')
+'user': Path(dirs.user_config_dir),
-read_config = deepcopy(config)
+'system': Path(dirs.site_config_dir),
-if 'libraries' not in config:
+}
-config['libraries'] = ['/path/to/ldraw']
-if config != read_config:
+def ldraw_dirs_from_config():
-config.write()
+libraries = []
-check_library_paths(config)
+dirs = config_dirs()
-load_ldconfig_ldr(config)
+for dirpath in [dirs['system'], dirs['user']]:
-return config
+filename = dirpath / 'ldcheck.cfg'
+from configobj import ConfigObj
-def library_paths(config):
+config = ConfigObj(str(filename), encoding = 'UTF8')
-for library_path_string in config['libraries']:
+if 'libraries' in config:
-yield Path(library_path_string).expanduser()
+libraries = expand_paths(config['libraries'])
+return libraries
-def check_library_paths(config):
-from sys import exit
+def expand_paths(paths):
-problems = False
+return [
-have_paths = False
+Path(library).expanduser()
-for library_path in library_paths(config):
+for library in paths
-have_paths = True
+]
-if not library_path.exists():
-problems = True
+class LDrawContext:
-print(str.format(
+'''
-'Library path {} does not exist',
+Contains context-dependant LDraw information, like library directory
-library_path,
+paths and the colour table.
-))
+'''
-elif not library_path.exists():
+def __init__(self, libraries = None):
-problems = True
+self._libraries = libraries
-print(str.format(
+if not self._libraries:
-'Library path {} is not a directory',
+self._libraries = ldraw_dirs_from_config()
-library_path,
+self.ldconfig_colour_data = self.load_ldconfig_ldr()
-))
+self.check_library_paths()
-if not have_paths:
+@property
-print('No LDraw path specified')
+def libraries(self):
-problems = True
+return self._libraries
-if problems:
+@property
-print('Please fix ldcheck.cfg')
+def colours(self):
-exit(1)
+return self.ldconfig_colour_data
+def ldconfig_ldr_discovery(self):
-def find_ldconfig_ldr_paths(config):
+for library_path in self.libraries:
-for library_path in library_paths(config):
+yield from [
-yield from [
+library_path / path
-library_path / path
+for path in ['LDConfig.ldr', 'ldconfig.ldr']
-for path in ['LDConfig.ldr', 'ldconfig.ldr']
+if (library_path / path).is_file()
-if (library_path / path).is_file()
+]
-]
+def load_ldconfig_ldr(self):
+from colours import load_colours
-import argparse
+for ldconfig_ldr_path in self.ldconfig_ldr_discovery():
+with open(ldconfig_ldr_path) as ldconfig_ldr:
+return load_colours(ldconfig_ldr)
+def check_library_paths(self):
+from sys import stderr
+problems = False
+have_paths = False
+for library_path in self.libraries:
+have_paths = True
+if not library_path.exists():
+problems = True
+print(str.format(
+'Library path {} does not exist',
+library_path,
+), file = stderr)
+elif not library_path.exists():
+problems = True
+print(str.format(
+'Library path {} is not a directory',
+library_path,
+), file = stderr)
+if not have_paths:
+raise RuntimeError('no LDraw library paths specified')
+def is_ldconfig_colour(self, colour):
+return colour.index in self.colours
+def colour_name(self, colour):
+if self.is_ldconfig_colour(colour):
+return self.colours[self.index]['name']
+else:
+return str(colour)
+def colour_face(self, colour):
+if self.is_ldconfig_colour(colour):
+return self.colours[self.index]['value']
+elif colour.is_direct_colour:
+return '#%06X' % (self.index & 0xffffff)
+else:
+return '#000000'
+def is_valid_colour(self, colour):
+return self.is_ldconfig_colour(colour) or colour.is_direct_colour
 class ListProblemsAction(argparse.Action):
 def __init__(self, option_strings, dest, nargs = None, **kwargs):
 super().__init__(option_strings, dest, nargs = 0, **kwargs)
 def __call__(self, *args, **kwargs):
 name = warning_type.name,
 severity = warning_type.severity,
 message = warning_type.placeholder_message(),
 ))
 exit(0)
-def load_ldconfig_ldr(config):
-for ldconfig_ldr_path in find_ldconfig_ldr_paths(config):
-with ldconfig_ldr_path.open() as ldconfig_ldr:
-load_colours(ldconfig_ldr)
 def format_report(report, model, test_suite, *, use_colors = True):
 from testsuite import problem_text
 messages = []
 for problem in report['problems']:
 )
 parser.add_argument('--color',
 action = 'store_true',
 help = 'use colors'
 )
+parser.add_argument('-d', '--ldraw-dir',
+nargs = '+',
+help = 'specify LDraw directory path(s)',
+)
 parser.add_argument('-v', '--version',
 action = 'version',
 version = str.format('{appname} {version}',
 appname = appname,
 version = version_string,
 ),
 )
 args = parser.parse_args()
-config = load_config()
+libraries = ldraw_dirs_from_config()
+if args.ldraw_dir:
+libraries = expand_paths(args.ldraw_dir)
+try:
+context = LDrawContext(libraries)
+except RuntimeError as error:
+print('error:', str(error), file = stderr)
+exit(1)
 if args.color:
 try:
 import colorama
 colorama.init()
 except ImportError:
 print('Use of --color requires the colorama module, disabling colours', file = stderr)
 args.color = False
 if args.subfile:
 import filecache
-cache = filecache.SubfileCache(
+cache = filecache.SubfileCache(context = context)
-ldraw_directories = config['libraries'],
-)
 subfile = cache.prepare_file(args.filename)
 if not subfile.valid:
 print(subfile.problem)
 else:
 print('Flat dimensions:', repr(subfile.flatness))
 with open(args.filename, 'rb') as file:
 from os.path import basename
 model = parse.read_ldraw(
 file,
 name = basename(args.filename),
-ldraw_directories = config['libraries'])
+context = context)
 if args.dump:
 print('header: ' + type(model.header).__name__)
 for key in sorted(dir(model.header)):
 if not key.startswith('__'):
 print('\t' + key + ': ' + repr(getattr(model.header, key)))

Mercurial > ldcheck / file comparison

comparison: ldcheck.py

ldcheck.py