feat: file move and mkdir

Author: easepan

fops.py

@@ -11,17 +11,22 @@
 # ------------------------------------------------------------------------- #
 # find,file_pattern:string,root:string
 # ------------------------------------------------------------------------- #
+# mkdir,folder:string
+# ------------------------------------------------------------------------- #
 
 import sys
 import os
 import re
+import shutil
 
 
 def file_operation(file_pattern: str, root: str, target: str, callback):
     for (dirpath, dirnames, filenames) in os.walk(root):
         for filename in filenames:
             path = '%s%s%s' % (dirpath, os.path.sep, filename)
             if re.search(file_pattern, path):
+                target = '%s%s%s' % (target, '' if target.endswith(
+                    os.path.sep) else os.path.sep, filename)
                 callback(path, target)
 
 
@@ -34,6 +39,15 @@ def callback_of_removing(path: str, target: str):
     os.remove(path)
 
 
+def callback_of_copying(path: str, target: str):
+    print('copy file %s to %s' % (path, target))
+    shutil.copyfile(path, target)
+    
+def callback_of_moving(path: str, target: str):
+    print('move file %s to %s' % (path, target))
+    shutil.move(path, target)
+
+
 # print(sys.argv)
 for arg in sys.argv[1:]:
     cmd_arg = str.split(arg, ',')[1:]
@@ -55,3 +69,9 @@ def callback_of_removing(path: str, target: str):
         file_operation(file_pattern, root, target, callback_of_removing)
     if str.startswith(arg, 'find'):
         file_operation(file_pattern, root, target, callback_of_finding)
+    if str.startswith(arg, 'cp'):
+        file_operation(file_pattern, root, target, callback_of_copying)
+    if str.startswith(arg, 'mv'):
+        file_operation(file_pattern, root, target, callback_of_moving)
+    if str.startswith(arg, 'mkdir'):
+        os.makedirs(file_pattern)