Files
IfcOpenShell/src/ifc4d/__main__.py
T
2022-09-05 01:46:19 +01:00

80 lines
2.5 KiB
Python

import os
import sys
import argparse
from ifc4d.p6xer2ifc import P6XER2Ifc
from ifc4d.p62ifc import P62Ifc
from ifc4d.msp2ifc import MSP2Ifc
from ifc4d.pp2ifc import PP2Ifc
import ifcopenshell
parser = argparse.ArgumentParser()
parser .add_argument('-f','--file', action='store', type=str,
required=True, help="schedule file name to be parsed")
parser .add_argument('-s','--schedule', action='store', required=True,
type=str, help='file format as xer, p6xml, mspxml, pp')
parser .add_argument('-i', '--ifcfile', action='store', required=False,
type=str, help='ifc file name as string e.g. \"file.ifc\"')
parser .add_argument('-o', '--output', action='store', required=True,
type=str, help='ifc file name as string e.g. \"file.ifc\"')
args = parser .parse_args()
def get_file():
ifcfile = None
if args.ifcfile:
ifcfile = ifcopenshell.open(args.ifcfile)
elif args.output:
ifc = ifcopenshell.file(schema='IFC4')
ifc.create_entity("IfcWorkPlan")
ifc.create_entity("IfcProject")
ifc.write(args.output)
ifcfile = ifcopenshell.open(args.output)
else:
ifcfile = None
return ifcfile
if not args.ifcfile:
print("You need to provide an ifc file to add schedule")
print("Examples python ifc4d -i model.ifc -s xer -f schedule.xer -o newifc.ifc")
elif not args.output:
print("an output file is required to save changes")
print("python ifc4d -o newfile.ifc -s xer -f schedule.xer")
elif args.schedule== "xer":
p6xer = P6XER2Ifc()
p6xer.xer = args.file
p6xer.output = args.output
ifcfile = get_file()
if ifcfile:
p6xer.file = ifcfile
p6xer.execute()
else: raise Exception("No files provided for output")
elif args.schedule == "mspxml":
msp = MSP2Ifc()
msp.xml = args.file()
msp.output = args.output
ifcfile = get_file()
if ifcfile:
msp.file = ifcfile
msp.execute()
elif args.schedule == "p6xml":
p6xml = P62Ifc()
p6xml.output = args.output
p6xml.xml = args.file
ifcfile = get_file()
if ifcfile:
p6xml.file = ifcopenshell.open(args.ifcfile)
p6xml.execute()
elif args.schedule == "pp":
pp = PP2Ifc()
pp.output = args.output
pp.pp = args.file
ifcfile = get_file()
if ifcfile:
pp.file = ifcopenshell.open(args.ifcfile)
pp.execute()
else:
print("schedule type you selected is not implemented at the moment")