From d367127b4e2dc73174d9d40c6bf319c3335c6cf5 Mon Sep 17 00:00:00 2001
From: TING-JUN WANG <levi900227@gmail.com>
Date: Mon, 29 Apr 2024 22:30:48 +0800
Subject: [PATCH] style: split out the user list func

---
 src/users.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 src/users.py

diff --git a/src/users.py b/src/users.py
new file mode 100644
index 0000000..1b9bbbb
--- /dev/null
+++ b/src/users.py
@@ -0,0 +1,36 @@
+import os
+
+def get_master():
+    ans = []
+    for year in os.listdir(f'/volume2/m3/master/'):
+        if os.path.isdir(f'/volume2/m3/master/{year}'):
+            for user in os.listdir(f'/volume2/m3/master/{year}'):
+                ans.append(f'/volume2/m3/master/{year}/{user}')
+    for user in os.listdir(f'/volume2/m3/master08'):
+        ans.append(f'/volume2/m3/master08/{user}')
+    for user in os.listdir(f'/volume2/m3/master09'):
+        ans.append(f'/volume2/m3/master09/{user}')
+    return ans
+
+def get_phd():
+    ans = []
+    for dir_name in os.listdir(f'/volume2/m3/phd/'):
+        if dir_name.isnumeric():
+            year = dir_name
+            for user in os.listdir(f'/volume2/m3/phd/{year}'):
+                ans.append(f'/volume2/m3/phd/{year}/{user}')
+        else:
+            ans.append(f'/volume2/m3/phd/{dir_name}')
+    return ans
+
+def get_ra():
+    ans = []
+    for user in os.listdir(f'/volume2/m3/ra/'):
+        ans.append(f'/volume2/m3/ra/{user}')
+    return ans
+
+def get_extra():
+    ans = []
+    for user in os.listdir(f'/volume2/m3/extra/'):
+        ans.append(f'/volume2/m3/extra/{user}')
+    return ans