eden

Reviewed By: zsol

Differential Revision: D50490765

fbshipit-source-id: b11d0c956f4e84badf1d67a09e754a2b850aeac7

addons/TARGETS

@@ -7,6 +7,7 @@ oncall("scm_client_infra")
 
 python_binary(
     name = "build-tar-script",
+    main_function = ".build-tar.main",
     main_src = "build-tar.py",
 )
 

ci/TARGETS

@@ -9,7 +9,7 @@ python_binary(
     srcs = [
         "gen_workflows.py",
     ],
-    main_module = "eden.oss.ci.gen_workflows",
+    main_function = "eden.oss.ci.gen_workflows.main",
     deps = [
         "fbsource//third-party/pypi/pyyaml:pyyaml",
     ],

eden/facebook/TARGETS

@@ -7,7 +7,7 @@ oncall("scm_client_infra")
 python_binary(
     name = "fb-eden-tw-clone",
     srcs = ["fb_eden_tw_clone.py"],
-    main_module = "eden.facebook.fb_eden_tw_clone",
+    main_function = "eden.facebook.fb_eden_tw_clone.main_2",
 )
 
 make_rule_compatible_with(

eden/fs/cli/TARGETS

@@ -45,7 +45,7 @@ buck_command_alias(
 python_binary(
     name = "edenfsctl-with-privhelper",
     srcs = [],
-    main_module = "eden/fs/cli/main",
+    main_function = "eden.fs.cli.main.zipapp_main",
     deps = [
         ":main_lib",
         "//eden/fs/privhelper:privhelper",
@@ -55,7 +55,7 @@ python_binary(
 python_binary(
     name = "edenfsctl",
     srcs = [],
-    main_module = "eden/fs/cli/main",
+    main_function = "eden.fs.cli.main.zipapp_main",
     deps = [
         ":main_lib",
     ],
@@ -64,7 +64,7 @@ python_binary(
 python_binary(
     name = "repl",
     srcs = ["repl.py"],
-    main_module = "eden/fs/cli/repl",
+    main_function = "eden.fs.cli.repl.main",
     deps = [
         "fbsource//third-party/pypi/ipython:ipython",
         ":main_lib",  # @manual

eden/fs/cli/facebook/TARGETS

@@ -8,7 +8,7 @@ python_binary(
     compatible_with = [
         "ovr_config//os:linux",
     ],
-    main_module = "eden.fs.cli.facebook.trace",
+    main_function = "eden.fs.cli.facebook.trace.dump_to_artillery",
     deps = [
         ":trace",
     ],
@@ -40,7 +40,7 @@ python_library(
 
 python_binary(
     name = "test_hostcaps",
-    main_module = "eden.fs.cli.facebook.hostcaps",
+    main_function = "eden.fs.cli.facebook.hostcaps.main",
     deps = [":hostcaps"],
 )
 

eden/fs/cli_rs/TARGETS

@@ -9,7 +9,7 @@ oncall("scm_client_infra")
 python_binary(
     name = "gen_wrapper_location",
     srcs = ["facebook/gen_wrapper_location.py"],
-    main_module = "eden/fs/cli_rs/facebook/gen_wrapper_location",
+    main_function = "eden.fs.cli_rs.facebook.gen_wrapper_location.main_2",
 )
 
 buck_genrule(

eden/fs/config/facebook/TARGETS

@@ -23,7 +23,7 @@ python_library(
 python_binary(
     name = "edenfs_config_manager",
     srcs = ["edenfs_config_manager.py"],
-    main_module = "eden.fs.config.facebook.edenfs_config_manager",
+    main_function = "eden.fs.config.facebook.edenfs_config_manager.main",
     deps = [
         ":edenfs_config_manager_lib",
     ],

eden/fs/facebook/TARGETS

@@ -51,7 +51,7 @@ python_binary(
     # @autodeps-skip
     name = "edenfs_restarter",
     srcs = ["edenfs_restarter.py"],
-    main_module = "eden.fs.facebook.edenfs_restarter",
+    main_function = "eden.fs.facebook.edenfs_restarter.main",
     os_deps = [(
         "linux",
         ["//eden/fs/cli/facebook:telemetry"],
@@ -68,13 +68,13 @@ python_binary(
     # @autodeps-skip
     name = "make-release",
     srcs = ["make-release.py"],
-    main_module = "eden.fs.facebook.make-release",
+    main_function = "eden.fs.facebook.make-release.main_2",
 )
 
 python_binary(
     name = "fix-windows-eden",
     srcs = ["fix-windows-eden.py"],
-    main_module = "eden.fs.facebook.fix-windows-eden",
+    main_function = "eden.fs.facebook.fix-windows-eden.main_2",
     deps = [
         "fbsource//third-party/pypi/colorama:colorama",
         "fbsource//third-party/pypi/pip:pip",

eden/fs/facebook/prefetch_profiles/TARGETS

@@ -19,7 +19,7 @@ python_binary(
     compatible_with = [
         "ovr_config//os:linux",
     ],
-    main_module = "eden.fs.facebook.prefetch_profiles.generate_fbandroid_specs",
+    main_function = "eden.fs.facebook.prefetch_profiles.generate_fbandroid_specs.main",
     deps = [
         ":prefetch_profile_util",
         "//libfb/py:employee",
@@ -34,7 +34,7 @@ python_binary(
     compatible_with = [
         "ovr_config//os:linux",
     ],
-    main_module = "eden.fs.facebook.prefetch_profiles.generate_fbobjc_specs",
+    main_function = "eden.fs.facebook.prefetch_profiles.generate_fbobjc_specs.main",
     deps = [
         ":prefetch_profile_util",
         "//libfb/py:employee",
@@ -68,7 +68,7 @@ python_binary(
     compatible_with = [
         "ovr_config//os:linux",
     ],
-    main_module = "eden.fs.facebook.prefetch_profiles.glob_filenames",
+    main_function = "eden.fs.facebook.prefetch_profiles.glob_filenames.main",
     deps = [
         ":glob_filenames_main",
     ],
@@ -88,7 +88,7 @@ python_library(
 python_binary(
     name = "prefetch_profile",
     srcs = ["record_prefetch_profile.py"],
-    main_module = "eden.fs.facebook.prefetch_profiles.record_prefetch_profile",
+    main_function = "eden.fs.facebook.prefetch_profiles.record_prefetch_profile.main",
     deps = [
         ":prefetch_profile_lib",
     ],
@@ -101,7 +101,7 @@ python_binary(
         "ovr_config//os:linux",
         "ovr_config//os:windows",
     ],
-    main_module = "eden.fs.facebook.prefetch_profiles.generate_sandcastle_jobs",
+    main_function = "eden.fs.facebook.prefetch_profiles.generate_sandcastle_jobs.main",
     deps = [
         ":generate_sandcastle_jobs_lib",
     ],
@@ -127,7 +127,7 @@ python_binary(
     compatible_with = [
         "ovr_config//os:linux",
     ],
-    main_module = "eden.fs.facebook.prefetch_profiles.profile_name_configerator_sync",
+    main_function = "eden.fs.facebook.prefetch_profiles.profile_name_configerator_sync.main",
     deps = [
         ":prefetch_profile_util",
         "//configerator/configo/client/thrift_py3:client",

eden/integration/snapshot/TARGETS

@@ -24,7 +24,7 @@ python_library(
 python_binary(
     name = "gen_snapshot",
     srcs = ["gen_snapshot.py"],
-    main_module = "eden.integration.snapshot.gen_snapshot",
+    main_function = "eden.integration.snapshot.gen_snapshot.main_2",
     deps = [
         ":snapshot",
         "//eden/integration/lib:lib",
@@ -34,7 +34,7 @@ python_binary(
 python_binary(
     name = "unpack_snapshot",
     srcs = ["unpack_snapshot.py"],
-    main_module = "eden.integration.snapshot.unpack_snapshot",
+    main_function = "eden.integration.snapshot.unpack_snapshot.main",
     deps = [
         ":snapshot",
     ],

eden/integration/snapshot/gen_snapshot.py

@@ -71,6 +71,10 @@ def main() -> int:
     return 0
 
 
-if __name__ == "__main__":
+def main_2() -> None:
     rc: int = main()
     sys.exit(rc)
+
+
+if __name__ == "__main__":
+    main_2()

eden/mononoke/facebook/backup_verifier/TARGETS

@@ -8,7 +8,7 @@ python_binary(
     srcs = [
         "main.py",
     ],
-    main_module = "eden.mononoke.facebook.backup_verifier.main",
+    main_function = "eden.mononoke.facebook.backup_verifier.main.main_2",
     deps = [
         "//configerator/distribution/api/py:configerator_config",
         "//configerator/structs/scm/mononoke/repos:repos-py",

eden/mononoke/facebook/blobimportjob/TARGETS

@@ -5,7 +5,7 @@ oncall("mononoke")
 python_binary(
     name = "blobimportjob",
     srcs = ["main.py"],
-    main_module = "eden.mononoke.facebook.blobimportjob.main",
+    main_function = "eden.mononoke.facebook.blobimportjob.main.main",
     deps = [
         "//configerator/distribution/api/py:configerator_config",
         "//configerator/structs/scm/mononoke/blobimport:blobimport_state-py",

eden/mononoke/facebook/blobstore_deletion/TARGETS

@@ -5,7 +5,7 @@ oncall("mononoke")
 python_binary(
     name = "manifold_copy",
     srcs = ["manifold_copy.py"],
-    main_module = "eden.mononoke.facebook.blobstore_deletion.manifold_copy",
+    main_function = "eden.mononoke.facebook.blobstore_deletion.manifold_copy.main",
     deps = [
         "//manifold/clients/python:manifold_client",
     ],
@@ -14,7 +14,7 @@ python_binary(
 python_binary(
     name = "manifold_key_download",
     srcs = ["manifold_key_download.py"],
-    main_module = "eden.mononoke.facebook.blobstore_deletion.manifold_key_download",
+    main_function = "eden.mononoke.facebook.blobstore_deletion.manifold_key_download.main",
     deps = [
         "//datainfra/presto/py:client_lib",
     ],

eden/mononoke/facebook/fbsource_commit_tailer/TARGETS

@@ -6,7 +6,7 @@ oncall("mononoke")
 python_binary(
     name = "fbsource_commit_tailer",
     srcs = glob(["*.py"]),
-    main_module = "eden.mononoke.facebook.fbsource_commit_tailer.tailer",
+    main_function = "eden.mononoke.facebook.fbsource_commit_tailer.tailer.main",
     deps = [
         "fbsource//third-party/pypi/aiohttp:aiohttp",
         "//eden/mononoke/facebook/mononoke_config:mononoke_config",

eden/mononoke/facebook/get_bundle/TARGETS

@@ -6,7 +6,7 @@ oncall("mononoke")
 python_binary(
     name = "get_bundle",
     srcs = ["get_bundle.py"],
-    main_module = "eden.mononoke.facebook.get_bundle.get_bundle",
+    main_function = "eden.mononoke.facebook.get_bundle.get_bundle.get_bundle",
     deps = [
         "fbsource//third-party/pypi/click:click",
         ":bundle_store_lib",

eden/mononoke/facebook/locked_repo_reporter/TARGETS

@@ -5,7 +5,7 @@ oncall("mononoke")
 python_binary(
     name = "locked_repo_reporter",
     srcs = ["locked_repo_reporter.py"],
-    main_module = "eden.mononoke.facebook.locked_repo_reporter.locked_repo_reporter",
+    main_function = "eden.mononoke.facebook.locked_repo_reporter.locked_repo_reporter.main",
     deps = [
         "//common/db/smc_db:py",
         "//configerator/distribution/api/py:configerator_config",

eden/mononoke/facebook/monitoring/TARGETS

@@ -10,7 +10,7 @@ python_binary(
     srcs = [
         "review_bookmarks.py",
     ],
-    main_module = "eden.mononoke.facebook.monitoring.review_bookmarks",
+    main_function = "eden.mononoke.facebook.monitoring.review_bookmarks.main_2",
     deps = [
         ":lib",
         "//libfb/py:log",

eden/mononoke/facebook/mononoke_config/TARGETS

@@ -20,7 +20,7 @@ python_binary(
     srcs = [
         "mononoke_find_repo_id.py",
     ],
-    main_module = "eden.mononoke.facebook.mononoke_config.mononoke_find_repo_id",
+    main_function = "eden.mononoke.facebook.mononoke_config.mononoke_find_repo_id.main_2",
     deps = [
         ":mononoke_config",
     ],

eden/mononoke/facebook/mononoke_pack_similar_blobs_download/TARGETS

@@ -5,7 +5,7 @@ oncall("mononoke")
 python_binary(
     name = "hive_similar_blobs_downloader",
     srcs = ["hive_similar_blobs_downloader.py"],
-    main_module = "eden.mononoke.facebook.mononoke_pack_similar_blobs_download.hive_similar_blobs_downloader",
+    main_function = "eden.mononoke.facebook.mononoke_pack_similar_blobs_download.hive_similar_blobs_downloader.main",
     deps = [
         "//datainfra/presto/py:client_lib",
     ],

eden/mononoke/facebook/post_commit_tailer/TARGETS

@@ -22,7 +22,7 @@ python_binary(
     srcs = [
         "example_tailer.py",
     ],
-    main_module = "eden.mononoke.facebook.post_commit_tailer.example_tailer",
+    main_function = "eden.mononoke.facebook.post_commit_tailer.example_tailer.main",
     deps = [
         ":post_commit_tailer",
     ],

eden/mononoke/facebook/replay/TARGETS

@@ -7,7 +7,7 @@ cogwheel_test(
     srcs = [
         "EdenAPIReplayTest.py",
     ],
-    main_module = "eden.mononoke.facebook.replay.EdenAPIReplayTest",
+    main_function = "eden.mononoke.facebook.replay.EdenAPIReplayTest.main",
     workload_config = create_cogwheel_config(
         ab_test = True,
         configerator_path_prefix_triggers = [

eden/mononoke/facebook/reposynthesizer/TARGETS

@@ -5,7 +5,7 @@ oncall("mononoke")
 python_binary(
     name = "reposynthesizer",
     srcs = ["reposynthesizer.py"],
-    main_module = "eden.mononoke.facebook.reposynthesizer.reposynthesizer",
+    main_function = "eden.mononoke.facebook.reposynthesizer.reposynthesizer.main",
     deps = [
         "//scm/lib:repo",
     ],

eden/mononoke/facebook/sparse_size_commit_tailer/TARGETS

@@ -8,7 +8,7 @@ python_binary(
     srcs = [
         "tailer.py",
     ],
-    main_module = "eden.mononoke.facebook.sparse_size_commit_tailer.tailer",
+    main_function = "eden.mononoke.facebook.sparse_size_commit_tailer.tailer.main",
     deps = [
         "//configerator/distribution/api/py:configerator_config",
         "//configerator/structs/scm/mononoke/repos:repos-py",

eden/mononoke/facebook/sqlblob_gc_py/TARGETS

@@ -7,7 +7,7 @@ python_binary(
     srcs = glob([
         "*.py",
     ]),
-    main_module = "eden.mononoke.facebook.sqlblob_gc_py.main",
+    main_function = "eden.mononoke.facebook.sqlblob_gc_py.main.main",
     deps = [
         "//configerator/common:crypto_utils",
         "//configerator/configo/client/thrift_py3:client",

eden/mononoke/tests/fixtures/facebook/generate_new_fixtures/TARGETS

@@ -5,7 +5,7 @@ oncall("mononoke")
 python_binary(
     name = "generate_new_fixtures",
     srcs = ["main.py"],
-    main_module = "eden.mononoke.tests.fixtures.facebook.generate_new_fixtures.main",
+    main_function = "eden.mononoke.tests.fixtures.facebook.generate_new_fixtures.main.main",
     deps = [
         "//scm/lib:repo",
     ],

eden/mononoke/tests/integration/TARGETS

@@ -40,7 +40,7 @@ python_binary(
     # @autodeps-skip
     name = "integration_runner_real",
     srcs = ["integration_runner_real.py"],
-    main_module = "eden.mononoke.tests.integration.integration_runner_real",
+    main_function = "eden.mononoke.tests.integration.integration_runner_real.run",
     deps = [
         "fbsource//third-party/pypi/click:click",
         "//eden/mononoke/tests/integration/facebook:lib_buck",

eden/mononoke/tests/integration/facebook/TARGETS

@@ -49,7 +49,7 @@ buck_filegroup(
 python_binary(
     name = "generate_manifest",
     srcs = ["generate_manifest.py"],
-    main_module = "eden.mononoke.tests.integration.facebook.generate_manifest",
+    main_function = "eden.mononoke.tests.integration.facebook.generate_manifest.main",
     deps = [
         ":lib_buck",
     ],

eden/mononoke/tests/utils/TARGETS

@@ -9,7 +9,7 @@ python_binary(
     srcs = [
         "generate_buck_target_deps.py",
     ],
-    main_module = "eden.mononoke.tests.utils.generate_buck_target_deps",
+    main_function = "eden.mononoke.tests.utils.generate_buck_target_deps.main",
 )
 
 rust_library(

eden/scm/hgcache_size_logger/TARGETS

@@ -8,7 +8,7 @@ python_binary(
         "hgcache_size_logger.py",
     ],
     base_module = "eden.scm",
-    main_module = "eden.scm.hgcache_size_logger",
+    main_function = "eden.scm.hgcache_size_logger.main",
     deps = [
         "//libfb/py:log",
         "//rfe/scubadata:scubadata_py_DEPRECATED_USE_py3_INSTEAD",

eden/scm/tests/TARGETS

@@ -13,7 +13,7 @@ python_binary(
     name = "run_tests",
     srcs = ["unittestify.py"] + glob(["**/*.t"]),
     base_module = "",
-    main_module = "unittestify",
+    main_function = "unittestify.main",
     package_style = "inplace",
     resources = {
         "//eden/scm/tests:dummyssh3": "dummyssh3.par",

eden/scm/tests/unittestify.py

@@ -178,8 +178,12 @@ class hgtests(unittest.TestCase):
                 setattr(cls, method_name, gettestmethod(name, port))
 
 
-if __name__ == "__main__":
+def main() -> None:
     args, env = prepareargsenv(os.getcwd())
     os.execvpe(args[0], args + sys.argv[1:], env)
+
+
+if __name__ == "__main__":
+    main()
 else:
     hgtests.collecttests(os.environ.get("HGTEST_DIR", "."))