export

oml/user/models.py

@@ -3,7 +3,10 @@
 from datetime import datetime
 import json
 import hashlib
+import os
+import shutil
 
+import ox
 import sqlalchemy as sa
 
 from changelog import Changelog
@@ -13,6 +16,8 @@ import json_pickler
 import settings
 import state
 import utils
+import media
+from websocket import trigger_event
 
 import logging
 logger = logging.getLogger(__name__)
@@ -323,6 +328,87 @@ class List(db.Model):
         state.db.session.add(self)
         state.db.session.commit()
 
+    def create_symlinks(self):
+        pass
+
+    @classmethod
+    def export(cls, data):
+        with db.session():
+            self = cls.get(data['list'])
+            if not self:
+                return
+            mode = data.get('mode')
+            prefix = data.get('path')
+            if mode not in ('add', 'replace'):
+                logger.debug('invalid mode %s', mode)
+                return
+            if not prefix or prefix == '/':
+                logger.debug('invalid export path %s', prefix)
+                trigger_event('activity', {
+                    'activity': 'export',
+                    'path': prefix,
+                    'progress': [0, 0],
+                    'status': {'code': 404, 'text': 'invalid export path'}
+                })
+                return
+            root = prefix
+            while not os.path.exists(root) and root != '/':
+                root = os.path.dirname(root)
+            if not os.access(root, os.W_OK):
+                logger.debug('can not write to %s', root)
+                trigger_event('activity', {
+                    'activity': 'export',
+                    'path': prefix,
+                    'progress': [0, 0],
+                    'path': prefix,
+                    'status': {'code': 404, 'text': 'permission denied'}
+                })
+                return
+            if os.path.exists(prefix):
+                existing_files = set(
+                    os.path.join(root, f) for root, _, files in os.walk(prefix) for f in files
+                )
+            else:
+                existing_files = set()
+            new_files = set()
+            count = self.get_items().count()
+            n = 1
+            for i in self.get_items():
+                if i.files.all():
+                    f = i.files.all()[0]
+                    source = f.fullpath()
+                    target = os.path.join(prefix, f.path)
+                    if mode == 'add':
+                        p = 1
+                        parts = target.rsplit('.', 1)
+                        while os.path.exists(target) and media.get_id(target) != f.sha1:
+                            target = '.'.join([parts[0], f.sha1[:p], parts[1]])
+                            p += 1
+                    ox.makedirs(os.path.dirname(target))
+                    if os.path.exists(target):
+                        if mode == 'replace' and media.get_id(target) != f.sha1:
+                            os.unlink(target)
+                            shutil.copy2(source, target)
+                    else:
+                        shutil.copy2(source, target)
+                    new_files.add(target)
+                trigger_event('activity', {
+                    'activity': 'export',
+                    'path': prefix,
+                    'progress': [n, count]
+                })
+                n += 1
+            if mode == 'replace':
+                for f in list(existing_files - new_files):
+                    os.unlink(f)
+                utils.remove_empty_folders(prefix)
+            trigger_event('activity', {
+                'activity': 'export',
+                'progress': [count, count],
+                'path': prefix,
+                'status': {'code': 200, 'text': ''},
+            })
+
 class Metadata(db.Model):
     __tablename__ = 'user_metadata'