feat(flow-python): add version (#56)

Co-authored-by: MegEngine <[email protected]>

Author: tpoisonooo

flow-debugger/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "flow-debugger"
-version = "0.1.0"
+version = "0.3.5"
 edition = "2018"
 authors = ["megvii"]
 

flow-derive/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "flow-derive"
-version = "0.3.2"
+version = "0.3.5"
 authors = ["megvii"]
 edition = "2018"
 

flow-message/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "flow-message"
-version = "0.3.2"
+version = "0.3.5"
 edition = "2018"
 
 [features]

flow-plugins/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "flow-plugins"
-version = "0.3.2"
+version = "0.3.5"
 authors = ["megvii"]
 edition = "2018"
 

flow-python/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "flow-python"
-version = "0.3.2"
+version = "0.3.5"
 authors = ["megvii"]
 edition = "2018"
 

flow-python/examples/basis_function/nest/nest.toml

@@ -22,23 +22,23 @@ connections = [
 
 [[graphs.nodes]]
 name = "n1"
-ty = "noop"
+ty = "Noop"
 
 [[graphs.nodes]]
 name = "n2"
 ty = "noop"
 
 [[graphs.nodes]]
 name = "n3"
-ty = "noop"
+ty = "Noop"
 
 [[graphs.nodes]]
 name = "n4"
 ty = "noop"
 
 [[graphs.nodes]]
 name = "n5"
-ty = "noop"
+ty = "Noop"
 
 [[graphs]]
 name = "nest"

flow-python/examples/basis_function/nest/op.py

@@ -4,17 +4,18 @@
 
 @map_def(exclusive=True)
 def noop(x):
-    print('send', x)
+    # print('send', x)
     return x
 
 
 @source_def()
 def gen(context=None):
     for i in range(context.n):
-        print('first', i)
+        # print('first', i)
         yield i
 
 
 @sink_def()
 def printer(x):
-    print('last', x)
+    pass
+    # print('last', x)

flow-python/megflow/__init__.py

@@ -11,3 +11,5 @@
 
 from .registry import register, collect, res_register
 from .megflow import *
+
+__version__ = version()

flow-python/megflow/command_line.py

@@ -23,18 +23,19 @@ def megflow_run():
                 sys.exit(1)
 
     import argparse
-    from megflow import Graph
+    import megflow
 
     parser = argparse.ArgumentParser(prog='megflow_run', description='run a pipeline with plugins.')
     parser.add_argument('--dump', help='the path to dump graph', action='store_true')
     parser.add_argument('-p', '--plugin', required=True, type=str, help='plugin path')
     parser.add_argument('-m', '--module', type=str, help='module path')
     parser.add_argument('-c', '--config', type=str, help='config path')
     parser.add_argument('--dynamic', type=str, help='dynamic config path')
+    parser.add_argument('--version', action='version', version='%(prog)s {version}'.format(version=megflow.__version__))
 
     args = parser.parse_args()
 
-    Graph(
+    megflow.Graph(
         dump=args.dump, 
         plugin_path=args.plugin, 
         module_path=args.module, 

flow-python/setup.py

@@ -30,12 +30,9 @@
 elif system == 'windows':
     dyn_ext = 'dll'
 
-devel_version = os.environ.get("DEVEL_VERSION")
 debug = os.environ.get("DEBUG")
 target_dir = os.environ.get("CARGO_TARGET_DIR")
 
-if not devel_version:
-    devel_version = "0.3.2"  # fall back
 if not debug:
     debug = False
 if not target_dir:
@@ -120,14 +117,22 @@ def run(self):
         pattern = re.compile(f'.*?{dyn_ext}\.[0-9]*')
         ext_modules.append(CopyExtension(pattern, f"{ffmpeg_dir}/lib/*.{dyn_ext}.*", "lib/"))
 
+    current_dir = os.getcwd()
+    with open(current_dir+'/Cargo.toml') as f:
+        pattern = re.compile(r'\d+\.(?:\d+\.)*\d+')
+        for line in f:
+            if line.startswith('version'):
+                version = re.search(pattern, line).group()
+                break
+
     setup(
         options={
             'bdist_wheel': {
                 'py_limited_api': "cp36",
             }
         },
         name="megflow",
-        version=devel_version,
+        version=version,
         packages=["megflow"],
         author="Megvii IPU-SDK Team",
         author_email="[email protected]",