From 0b59ff1a1656e08d75fc3e43b069dffd0b0338db Mon Sep 17 00:00:00 2001
From: wu_xinjun <bent20140204@outlook.com>
Date: 星期一, 14 三月 2022 14:30:37 +0800
Subject: [PATCH] Merge a new feature into dev

---
 src/utils/udp_get.py |   17 ++++++++++++-----
 1 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/src/utils/udp_get.py b/src/utils/udp_get.py
index a34d386..a01d4c4 100644
--- a/src/utils/udp_get.py
+++ b/src/utils/udp_get.py
@@ -25,7 +25,10 @@
             start = datetime.datetime.now()
             if not os.path.exists(save_folder):
                 os.mkdir(save_folder)
-            path = os.path.join(save_folder,f"{device_id}-{start.strftime('%Y-%m-%d-%H-%M-%S')}")
+            if sub_folder is not None:
+                path = os.path.join(save_folder,sub_folder)
+            else:
+                path = os.path.join(save_folder,f"{device_id}-{start.strftime('%Y-%m-%d-%H-%M-%S')}")
             if not os.path.exists(path):
                 os.mkdir(path)
             path = os.path.abspath(path)
@@ -46,17 +49,19 @@
 if __name__ == '__main__':
 
     def help():
-        print("Usage: udp_get.py [MICROWAVEhost] [PACKETthreshold] [DEVICEname] [SAVEfolder]")
-        print("EXAMPLE: udp_get.py 192.168.2.65 1000 pod1 ./manual_ply")
+        print("**************************ERROR**************************")
+        print("Usage: udp_get.py [MICROWAVEhost] [PACKETthreshold] [DEVICEname] [SAVEfolder] [SUBfolder")
+        print("EXAMPLE: udp_get.py 192.168.2.65 1000 pod1 ./manual_ply 60 \n")
         exit(1)
-
-    if len(sys.argv) != 5:
+        
+    if len(sys.argv) != 6:
         help()
     try:
         MICROWAVEhost = str(sys.argv[1])
         PACKETthreshold = int(sys.argv[2])
         DEVICEname = str(sys.argv[3])
         SAVEfolder = str(sys.argv[4])
+        SUBfolder = str(sys.argv[5])
 
         global packet_threshold
         packet_threshold = PACKETthreshold
@@ -64,6 +69,8 @@
         device_id = DEVICEname
         global save_folder
         save_folder = SAVEfolder
+        global sub_folder
+        sub_folder = SUBfolder
 
     except Exception as e:
         print(e)

--
Gitblit v1.9.1