64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
|
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])
|