diff --git a/aman/setup.py b/setup.py similarity index 76% rename from aman/setup.py rename to setup.py index 15e4ddf..d3e5af7 100644 --- a/aman/setup.py +++ b/setup.py @@ -11,7 +11,7 @@ from distutils.command.build_py import build_py_2to3 as _build_py # @brief Creates Protobuf python files to encode and decode messages # @param[in] source The protobuf source file def generateProtobuf(source): - output = source.replace('.proto', '_pb2.py').replace('../src/protobuf/', '') + output = source.replace('.proto', '_pb2.py').replace('src/protobuf/', '') if (not os.path.exists(output) or (os.path.exists(source) and os.path.getmtime(source) > os.path.getmtime(output))): print('Generating %s...' % output) @@ -20,11 +20,11 @@ def generateProtobuf(source): sys.stderr.write('Cannot find %s' % source) sys.exit(-1) - if not os.path.exists('../external/bin/protoc.exe'): + if not os.path.exists('external/bin/protoc.exe'): sys.stderr.write('Cannot find proto-compiler') sys.exit(-1) - command = [ '../external/bin/protoc.exe', '-I../src/protobuf/', '-I.', '--python_out=.', source] + command = [ 'external/bin/protoc.exe', '-Isrc/protobuf/', '-I.', '--python_out=.', source] if 0 != subprocess.call(command): sys.exit(-1) @@ -38,7 +38,7 @@ class clean(_clean): if filepath.endswith('_pb2.py') or filepath.endswith('.pyc'): os.remove(filepath) for dirname in dirnames: - if 'Arrival_MANager.egg-info' == dirname: + if 'Arrival_MANager.egg-info' == dirname or 'build' == dirname: shutil.rmtree(os.path.join(dirpath, dirname)) _clean.run(self) @@ -46,10 +46,10 @@ class clean(_clean): # @param[in] _build_py Instance of setuptools to build the system class build_py(_build_py): def run(self): - generateProtobuf('../src/protobuf/Aircraft.proto') - generateProtobuf('../src/protobuf/AircraftReport.proto') - generateProtobuf('../src/protobuf/AircraftSchedule.proto') - generateProtobuf('../src/protobuf/BaseTypes.proto') + generateProtobuf('src/protobuf/Aircraft.proto') + generateProtobuf('src/protobuf/AircraftReport.proto') + generateProtobuf('src/protobuf/AircraftSchedule.proto') + generateProtobuf('src/protobuf/BaseTypes.proto') _build_py.run(self) setup(