test

lib/tool_base/os_command.py

@@ -0,0 +1,42 @@
+import os, traceback, io, re, platform
+from . import logger
+from . import ToolSubprocess
+from . import ToolUtil, ToolBaseFile
+
+class ToolOSCommand(object):
+
+    @classmethod
+    def get_size(cls, path):
+        if platform.system() == 'Windows':
+            #https://docs.microsoft.com/en-us/sysinternals/downloads/du
+            """
+            bin = r'C:\SJVA3\data\bin\du64.exe'
+            command = [bin, '-c', '-nobanner', f'"{path}"']
+            data = ToolSubprocess.execute_command_return(command, force_log=True)
+            logger.warning(data)
+            ret = {}
+            tmp = data.split('\t')
+            ret['target'] = tmp[1].strip()
+            ret['size'] = int(tmp[0].strip())
+            ret['sizeh'] = ToolUtil.sizeof_fmt(ret['size'])
+            """
+            ret = {}
+            ret['target'] = path
+            if os.path.exists(path):
+                if os.path.isdir(path):
+                    ret['size'] = ToolBaseFile.size(start_path=path)
+                else:
+                    ret['size'] = os.stat(path).st_size
+            ret['sizeh'] = ToolUtil.sizeof_fmt(ret['size'])
+            return ret
+
+        else:
+            command = ['du', '-bs', path]
+            data = ToolSubprocess.execute_command_return(command)
+            ret = {}
+            tmp = data.split('\t')
+            ret['target'] = tmp[1].strip()
+            ret['size'] = int(tmp[0].strip())
+            ret['sizeh'] = ToolUtil.sizeof_fmt(ret['size'])
+        return ret
+