first commit

lib/framework/common/celery/shutil_task.py

@@ -0,0 +1,156 @@
+# -*- coding: utf-8 -*-
+import os
+import traceback
+import shutil
+
+from framework import app, celery, logger
+
+# run_in_celery=True 이미 celery안에서 실행된다. 바로 콜한다.
+def move(source_path, target_path, run_in_celery=False):
+    try:
+        if app.config['config']['use_celery'] and run_in_celery == False:
+            result = _move_task.apply_async((source_path, target_path))
+            return result.get()
+        else:
+            return _move_task(source_path, target_path)
+    except Exception as exception: 
+        logger.error('Exception:%s', exception)
+        logger.error(traceback.format_exc())
+        return _move_task(source_path, target_path)
+
+
+@celery.task
+def _move_task(source_path, target_path):
+    try:
+        logger.debug('_move_task:%s %s', source_path, target_path)
+        shutil.move(source_path, target_path)
+        logger.debug('_move_task end')
+        return True
+    except Exception as exception: 
+        logger.error('Exception:%s', exception)
+        logger.error(traceback.format_exc())
+        return False
+
+def move_exist_remove(source_path, target_path, run_in_celery=False):
+    try:
+        if app.config['config']['use_celery'] and run_in_celery == False:
+            result = _move_exist_remove_task.apply_async((source_path, target_path))
+            return result.get()
+        else:
+            return _move_exist_remove_task(source_path, target_path)
+    except Exception as exception: 
+        logger.error('Exception:%s', exception)
+        logger.error(traceback.format_exc())
+        return _move_exist_remove_task(source_path, target_path)
+
+@celery.task
+def _move_exist_remove_task(source_path, target_path):
+    try:
+        target_file_path = os.path.join(target_path, os.path.basename(source_path))
+        if os.path.exists(target_file_path):
+            os.remove(source_path)
+            return True
+        logger.debug('_move_exist_remove:%s %s', source_path, target_path)
+        shutil.move(source_path, target_path)
+        logger.debug('_move_exist_remove end')
+        return True
+    except Exception as exception: 
+        logger.error('Exception:%s', exception)
+        logger.error(traceback.format_exc())
+        return False
+
+
+def copytree(source_path, target_path):
+    try:
+        if app.config['config']['use_celery']:
+            result = _copytree_task.apply_async((source_path, target_path))
+            return result.get()
+        else:
+            return _copytree_task(source_path, target_path)
+    except Exception as exception: 
+        logger.error('Exception:%s', exception)
+        logger.error(traceback.format_exc())
+        return _copytree_task(source_path, target_path)
+
+@celery.task
+def _copytree_task(source_path, target_path):
+    try:
+        shutil.copytree(source_path, target_path)
+        return True
+    except Exception as exception: 
+        logger.error('Exception:%s', exception)
+        logger.error(traceback.format_exc())
+        return False
+
+# copy
+def copy(source_path, target_path):
+    try:
+        if app.config['config']['use_celery']:
+            result = _copy_task.apply_async((source_path, target_path))
+            return result.get()
+        else:
+            return _copy_task(source_path, target_path)
+    except Exception as exception: 
+        logger.error('Exception:%s', exception)
+        logger.error(traceback.format_exc())
+        return _copy_task(source_path, target_path)
+
+@celery.task
+def _copy_task(source_path, target_path):
+    try:
+        shutil.copy(source_path, target_path)
+        return True
+    except Exception as exception: 
+        logger.error('Exception:%s', exception)
+        logger.error(traceback.format_exc())
+        return False
+
+# rmtree
+def rmtree(source_path):
+    try:
+        if app.config['config']['use_celery']:
+            result = _rmtree_task.apply_async((source_path,))
+            return result.get()
+        else:
+            return _rmtree_task(source_path)
+    except Exception as exception: 
+        logger.error('Exception:%s', exception)
+        logger.error(traceback.format_exc())
+        return _rmtree_task(source_path)
+
+@celery.task
+def _rmtree_task(source_path):
+    try:
+        shutil.rmtree(source_path)
+        return True
+    except Exception as exception: 
+        logger.error('Exception:%s', exception)
+        logger.error(traceback.format_exc())
+        return False   
+
+
+def remove(remove_path):
+    try:
+        logger.debug('CELERY os.remove start : %s', remove_path)
+        if app.config['config']['use_celery']:
+            result = _remove_task.apply_async((remove_path,))
+            return result.get()
+        else:
+            return _remove_task(remove_path)
+    except Exception as exception: 
+        logger.error('Exception:%s', exception)
+        logger.error(traceback.format_exc())
+        return _remove_task(remove_path)
+    finally:
+        logger.debug('CELERY os.remove end : %s', remove_path)
+
+
+@celery.task
+def _remove_task(remove_path):
+    try:
+        os.remove(remove_path)
+        return True
+    except Exception as exception: 
+        logger.error('Exception:%s', exception)
+        logger.error(traceback.format_exc())
+        return False