From 056b3524e5849d9876846f4d7cdb7820da5ce892 Mon Sep 17 00:00:00 2001
From: Swann <swann.martinez@protonmail.com>
Date: Fri, 11 Dec 2020 23:02:20 +0100
Subject: [PATCH] feat: gzip to compress graph

---
 multi_user/delayable.py | 36 ++----------------------------------
 multi_user/operators.py | 39 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 41 insertions(+), 34 deletions(-)

diff --git a/multi_user/delayable.py b/multi_user/delayable.py
index 29bf282..f407d00 100644
--- a/multi_user/delayable.py
+++ b/multi_user/delayable.py
@@ -110,45 +110,13 @@ class Timer(Delayable):
         self.is_running = False
 
 class SessionRecordGraphTimer(Timer):
-    def __init__(self, timout=60, filepath=None):
+    def __init__(self, timout=10, filepath=None):
         self._filepath = filepath
         super().__init__(timout)
 
 
     def execute(self):
-        import networkx as nx
-        import pickle
-        import copy
-        from time import gmtime, strftime
-        from pathlib import Path
-
-        # Replication graph 
-        nodes_ids = session.list()
-        #TODO: add dump graph to replication
-
-        nodes =[]
-        for n in nodes_ids:
-            nd = session.get(uuid=n)
-            nodes.append((
-                n,
-                {
-                    'owner': nd.owner,
-                    'str_type': nd.str_type,
-                    'data': nd.data,
-                    'dependencies': nd.dependencies,
-                }
-            ))
-
-        db = dict()
-        db['nodes'] = nodes
-        db['users'] = copy.copy(session.online_users)
-
-        time = strftime("%Y_%m_%d_%H_%M_%S", gmtime())
-        filepath = Path(self._filepath)
-        filepath = filepath.with_name(f"{filepath.stem}_{time}{filepath.suffix}")
-        with open(filepath, "wb") as f:
-            logging.info(f"Writing db snapshot to {filepath}")
-            pickle.dump(db, f, protocol=4)
+        operators.dump_db(self._filepath)
 
 class ApplyTimer(Timer):
     def __init__(self, timout=1, target_type=None):
diff --git a/multi_user/operators.py b/multi_user/operators.py
index 28f141b..e02aca1 100644
--- a/multi_user/operators.py
+++ b/multi_user/operators.py
@@ -25,9 +25,11 @@ import shutil
 import string
 import sys
 import time
+from datetime import datetime
 from operator import itemgetter
 from pathlib import Path
 from queue import Queue
+import gzip
 
 import bpy
 import mathutils
@@ -714,6 +716,43 @@ class SessionNotifyOperator(bpy.types.Operator):
         return context.window_manager.invoke_props_dialog(self)
 
 
+def dump_db(filepath):
+    import networkx as nx
+    import pickle
+    import copy
+    from time import gmtime, strftime
+    from pathlib import Path
+
+    # Replication graph 
+    nodes_ids = session.list()
+    #TODO: add dump graph to replication
+
+    nodes =[]
+    for n in nodes_ids:
+        nd = session.get(uuid=n)
+        nodes.append((
+            n,
+            {
+                'owner': nd.owner,
+                'str_type': nd.str_type,
+                'data': nd.data,
+                'dependencies': nd.dependencies,
+            }
+        ))
+
+    db = dict()
+    db['nodes'] = nodes
+    db['users'] = copy.copy(session.online_users)
+
+    stime = datetime.now().strftime('%Y_%m_%d_%H-%M-%S')
+    
+    filepath = Path(filepath)
+    filepath = filepath.with_name(f"{filepath.stem}_{stime}{filepath.suffix}")
+    with gzip.open(filepath, "wb") as f:
+        logging.info(f"Writing db snapshot to {filepath}")
+        pickle.dump(db, f, protocol=4)
+
+
 class SessionRecordGraphOperator(bpy.types.Operator, ExportHelper):
     bl_idname = "session.export"
     bl_label = "SessionRecordGraph"