add dumpmap

src/c3nav/mapdata/packageio/write.py

@@ -0,0 +1,63 @@
+import json
+import os
+
+from django.conf import settings
+
+from ..models import Package
+from .utils import json_encode
+
+
+def write_packages(prettify=False):
+    print('Writing Map Packages…')
+    for package in Package.objects.all():
+        print('\n'+package.name)
+        write_package(package, prettify)
+
+
+def write_package(package, prettify=False):
+    pkg_path = os.path.join(settings.MAP_ROOT, package.directory)
+
+    with open(os.path.join(pkg_path, 'pkg.json'), 'w') as f:
+        f.write(json_encode(package.tofile()))
+
+    _write_folder(package, package.levels.all(), 'levels', prettify)
+    _write_folder(package, package.sources.all(), 'sources', prettify, check_sister_file=True)
+
+
+def _write_folder(package, objects, path, prettify=False, check_sister_file=False):
+    filenames = set()
+    full_path = os.path.join(settings.MAP_ROOT, package.directory, path)
+    if objects:
+        if not os.path.isdir(full_path):
+            os.mkdir(full_path)
+        for obj in objects:
+            filename = '%s.json' % obj.name
+            filenames.add(filename)
+
+            full_filename = os.path.join(full_path, filename)
+            new_data = obj.tofile()
+            new_data_encoded = json_encode(new_data)
+            if os.path.isfile(full_filename):
+                with open(full_filename) as f:
+                    old_data_encoded = f.read()
+                old_data = json.loads(old_data_encoded, parse_int=float)
+                if old_data != json.loads(new_data_encoded, parse_int=float):
+                    print('- Updated: '+os.path.join(path, filename))
+                elif old_data_encoded != new_data_encoded:
+                    if not prettify:
+                        continue
+                    print('- Beautified: '+os.path.join(path, filename))
+                else:
+                    continue
+            else:
+                print('- Created: '+os.path.join(path, filename))
+            with open(full_filename, 'w') as f:
+                f.write(new_data_encoded)
+
+    if os.path.isdir(path):
+        for filename in os.listdir(path):
+            full_filename = os.path.join(path, filename)
+            if filename not in filenames and filename.endswith('.json') and os.path.isfile(full_filename):
+                os.remove(full_filename)
+                if check_sister_file and os.path.isfile(full_filename[:-5]):
+                    os.remove(full_filename[:-5])