Build zipline-ai with Commands "zipline compile" and "zipline run"

api/py/ai/chronon/repo/explore.py

@@ -383,7 +383,8 @@ def load_team_data(path):
     "_events_without_topics": events_without_topics
 }
 
-if __name__ == "__main__":
+
+def main():
     parser = argparse.ArgumentParser(description="Explore tool for chronon")
     parser.add_argument("keyword", help="Keyword to look up keys")
     parser.add_argument("--conf-root", help="Conf root for the configs", default=CWD)
@@ -413,3 +414,7 @@ def load_team_data(path):
     else:
         group_bys = find_in_index(gb_index, args.keyword)
         display_entries(group_bys, args.keyword, root=root, trim_paths=True)
+
+
+if __name__ == "__main__":
+    main()

api/py/ai/chronon/repo/run.py

@@ -624,7 +624,7 @@ def set_defaults(parser):
     )
 
 
-if __name__ == "__main__":
+def main():
     parser = argparse.ArgumentParser(description="Submit various kinds of chronon jobs")
     parser.add_argument(
         "--conf",
@@ -722,3 +722,7 @@ def set_defaults(parser):
         )
     )
     Runner(args, os.path.expanduser(jar_path)).run()
+
+
+if __name__ == "__main__":
+    main()

api/py/setup.py

@@ -52,11 +52,17 @@ def get_version():
     ],
     long_description=long_description,
     long_description_content_type="text/markdown",
-    scripts=['ai/chronon/repo/explore.py', 'ai/chronon/repo/compile.py', 'ai/chronon/repo/run.py'],
+    entry_points={
+        "console_scripts": [
+            "compile=ai.chronon.repo.compile:extract_and_convert",
+            "explore=ai.chronon.repo.explore:main",
+            "run=ai.chronon.repo.run:main",
+        ]
+    },
     description="Chronon python API library",
     include_package_data=True,
     install_requires=basic_requirements,
-    name="chronon-ai",
+    name="zipline-ai",
     packages=find_packages(),
     extras_require={
         # Extra requirement to have access to cli commands in python2 environments.