浏览代码

added get_correct_csv_file()

Markus Spring 1 年之前
父节点
当前提交
bf07d23021
共有 1 个文件被更改,包括 9 次插入18 次删除
  1. 9 18
      apps/home/modules/utils.py

+ 9 - 18
apps/home/modules/utils.py

@@ -2,7 +2,8 @@
 utility functions for garmindb-mtb-dashboard
 """
 
-# import sys
+import os
+import sys
 
 from pathlib import Path
 
@@ -13,20 +14,10 @@ def is_running_in_docker():
     cgroup = Path('/proc/self/cgroup')
     return Path('/.dockerenv').is_file() or cgroup.is_file() and 'docker' in cgroup.read_text()
 
-# def is_running_in_docker():
-#     """
-#     Checks if the current process is running inside a Docker container.
-#     """
-#     try:
-#         with open('/proc/1/cgroup', 'r') as f:
-#             cgroup = f.read()
-#             return 'docker' in cgroup
-#     except FileNotFoundError:
-#         return False
-
-    # if is_running_in_docker():
-    #     sys.stderr.write("Running in a Docker container\n")
-    #     csv_file = '/root/HealthData/garmin_bike.csv'
-    # else:
-    #     sys.stderr.write("NOT Running in a Docker container\n" + os.getcwd() + "\n")
-    #     csv_file = './HealthData/garmin_bike.csv'
+def get_correct_csv_file():
+    if is_running_in_docker():
+        sys.stderr.write("Running in a Docker container\n")
+        return('/root/HealthData/garmin_bike.csv')
+    else:
+        sys.stderr.write("NOT Running in a Docker container\n" + os.getcwd() + "\n")
+        return('./HealthData/garmin_bike.csv')