mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-14 03:14:23 +00:00
johltn
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
# This can be packaged with `pyinstaller --onefile --clean --icon=icon.ico ifcpatch.py`
|
|
|
|
import ifcopenshell
|
|
import logging
|
|
import argparse
|
|
|
|
def execute(args, is_library=None):
|
|
logging.basicConfig(filename=args['log'], filemode='a', level=logging.DEBUG)
|
|
logger = logging.getLogger('IFCPatch')
|
|
print('# Loading IFC file ...')
|
|
ifc_file = ifcopenshell.open(args['input'])
|
|
print('# Loading patch recipe ...')
|
|
patcher = getattr(__import__('recipes.{}'.format(args['recipe'])), args['recipe']).Patcher(
|
|
args['input'], ifc_file, logger, args['arguments'])
|
|
print('# Patching ...')
|
|
patcher.patch()
|
|
ifc_file = patcher.file
|
|
if is_library is True:
|
|
return ifc_file
|
|
print('# Writing patched file ...')
|
|
if not args['output']:
|
|
args['output'] = args['input']
|
|
if isinstance(ifc_file, str):
|
|
with open(args['output'], 'w') as text_file:
|
|
text_file.write(ifc_file)
|
|
else:
|
|
ifc_file.write(args['output'])
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(
|
|
description='Patches IFC files to fix badly formatted data')
|
|
parser.add_argument(
|
|
'-i',
|
|
'--input',
|
|
type=str,
|
|
required=True,
|
|
help='The IFC file to patch')
|
|
parser.add_argument(
|
|
'-o',
|
|
'--output',
|
|
type=str,
|
|
help='The output file to save the patched IFC')
|
|
parser.add_argument(
|
|
'-r',
|
|
'--recipe',
|
|
type=str,
|
|
required=True,
|
|
help='Name of the recipe to use when patching')
|
|
parser.add_argument(
|
|
'-l',
|
|
'--log',
|
|
type=str,
|
|
help='Specify a log file',
|
|
default='ifcpatch.log')
|
|
parser.add_argument(
|
|
'-a',
|
|
'--arguments',
|
|
nargs='+',
|
|
help='Specify custom arguments to the patch recipe')
|
|
args = vars(parser.parse_args())
|
|
|
|
execute(args)
|
|
|
|
print('# All tasks are complete :-)')
|