Fix setup.py to keep sonic-pit files and directory.

Signed-off-by: Li Hua <guizhao.lh@alibaba-inc.com>
sonic-net · Oct 26, 2022 · 5ce634f · 5ce634f
1 parent 87af6a3
commit 5ce634f
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/sonic-pit/setup.py b/src/sonic-pit/setup.py
@@ -14,6 +14,20 @@
     maintainer_email='guizhao.lh@alibaba-inc.com',
     install_requires=dependencies,
     packages=find_packages(),
+    data_files=[
+        ('pit-sysdiag/src', ['pit-sysdiag/src/banner.py',
+                             'pit-sysdiag/src/errcode.py',
+                             'pit-sysdiag/src/sysdiag.py',
+                             'pit-sysdiag/src/pit_util_common.py',
+                             'pit-sysdiag/src/function.py',
+                             'pit-sysdiag/src/syslogger.py',
+                             'pit-sysdiag/src/test_case.py',
+                             'pit-sysdiag/src/psu_tc.py',
+                             'pit-sysdiag/src/fan_tc.py']),
+        ('pit-sysdiag/cases/fan_tc', ['pit-sysdiag/cases/fan_tc/config.json']),
+        ('pit-sysdiag/cases/psu_tc', ['pit-sysdiag/cases/psu_tc/config.json']),
+        ('pit-sysdiag/config/platform/x86_64-alibaba_as14-40d-cl-r0', ['pit-sysdiag/config/platform/x86_64-alibaba_as14-40d-cl-r0/case_config.json', 'pit-sysdiag/config/platform/x86_64-alibaba_as14-40d-cl-r0/platform_config.json']),
+    ],
     classifiers=[
         'Development Status :: 3 - Alpha',
         'Environment :: No Input/Output (Daemon)',