增加新文件

saturn_mount.sh

+import subprocess
+import shlex
+import sys
+import re
+from pathlib import Path
+
+blkid = "blkid"
+fstab = Path("/etc/fstab")
+mount = "mount -a"
+
+
+def runcmd(cmd, shell=False, encoding='utf-8'):
+    args = cmd if shell else shlex.split(cmd)
+    r = subprocess.run(args, shell=shell, stdout=subprocess.PIPE, stderr=subprocess.PIPE, encoding=encoding)
+    return r.stdout, r.returncode
+
+
+class edit_fstab:
+    def __init__(self, filename: Path, disks: dict):
+        self.filename = filename
+        self.disks = disks
+
+    def read_vars(self):
+        __vars = []
+        r = r'UUID=.*/opt/storage/data.*'
+        n = 1
+        with open(self.filename, 'r', encoding='utf-8', newline='') as f:
+            for line in f:
+                if re.match(r, line, re.M | re.I):
+                    continue
+                elif line:
+                    __vars.append(line.replace("\n", ""))
+        for uuid in self.disks.values():
+            line = "UUID={} /opt/storage/data{}     ext4     defaults,nofail 0 2".format(uuid, n)
+            n += 1
+            __vars.append(line)
+        return __vars
+
+    def write_vars(self, __vars: list):
+        with open(self.filename, 'w', encoding='utf-8', newline='') as f:
+            for i in __vars:
+                f.write("{}\n".format(i))
+
+
+def get_uuid(file: blkid, raid=False):
+    dev = "md" if raid else "sd"
+    r = r'(/dev/{}.*): UUID="(.*)" BLOCK_SIZE=.* TYPE="ext4".*PARTUUID=".*"'.format(dev)
+    disks = {}
+    with open(file, 'r', encoding='utf-8', newline='') as f:
+        for line in f:
+            mach = re.match(r, line, re.M | re.I)
+            if mach:
+                disks[mach.group(1)] = mach.group(2)
+    return disks
+
+
+if __name__ == '__main__':
+    raid = False
+    if len(sys.argv) == 2:
+        if sys.argv[1] == "--raid":
+            raid = True
+    disks = get_uuid(blkid, raid=raid)
+    if len(disks) in (1, 4, 8):
+        new_fstab = edit_fstab(fstab, disks)
+        new_fstab.write_vars(new_fstab.read_vars())
+        runcmd(mount)
\ No newline at end of file