sync_with_server.py
85 lines
| 3.1 KiB
| text/x-python
|
PythonLexer
|
r841 | import re | ||
|
r1144 | import xml.etree.ElementTree as ET | ||
|
r1321 | import httplib2 | ||
|
r841 | from django.core.management import BaseCommand | ||
|
r1321 | |||
|
r841 | from boards.models import GlobalId | ||
|
r1177 | from boards.models.post.sync import SyncManager | ||
|
r841 | |||
__author__ = 'neko259' | ||||
|
r1138 | REGEX_GLOBAL_ID = re.compile(r'(\w+)::([\w\+/]+)::(\d+)') | ||
|
r841 | |||
class Command(BaseCommand): | ||||
|
r1321 | help = 'Send a sync or get request to the server.' | ||
|
r841 | |||
|
r1138 | def add_arguments(self, parser): | ||
|
r1386 | parser.add_argument('url', type=str, help='Server root url') | ||
parser.add_argument('--global-id', type=str, default='', | ||||
|
r1321 | help='Post global ID') | ||
|
r1551 | parser.add_argument('--split-query', type=int, | ||
help='Split GET query into separate by the given' | ||||
' number of posts in one') | ||||
|
r1138 | |||
|
r841 | def handle(self, *args, **options): | ||
|
r1138 | url = options.get('url') | ||
|
r1386 | |||
pull_url = url + 'api/sync/pull/' | ||||
get_url = url + 'api/sync/get/' | ||||
|
r1511 | file_url = url[:-1] | ||
|
r1386 | |||
|
r1138 | global_id_str = options.get('global_id') | ||
if global_id_str: | ||||
match = REGEX_GLOBAL_ID.match(global_id_str) | ||||
if match: | ||||
key_type = match.group(1) | ||||
key = match.group(2) | ||||
local_id = match.group(3) | ||||
|
r841 | |||
|
r1138 | global_id = GlobalId(key_type=key_type, key=key, | ||
|
r841 | local_id=local_id) | ||
|
r1138 | xml = GlobalId.objects.generate_request_get([global_id]) | ||
|
r1177 | # body = urllib.parse.urlencode(data) | ||
|
r1138 | h = httplib2.Http() | ||
|
r1386 | response, content = h.request(get_url, method="POST", body=xml) | ||
|
r841 | |||
|
r1511 | SyncManager.parse_response_get(content, file_url) | ||
|
r1138 | else: | ||
raise Exception('Invalid global ID') | ||||
|
r841 | else: | ||
|
r1144 | h = httplib2.Http() | ||
|
r1321 | xml = GlobalId.objects.generate_request_pull() | ||
|
r1386 | response, content = h.request(pull_url, method="POST", body=xml) | ||
|
r1144 | |||
|
r1386 | print(content.decode() + '\n') | ||
|
r1144 | |||
root = ET.fromstring(content) | ||||
status = root.findall('status')[0].text | ||||
if status == 'success': | ||||
|
r1322 | ids_to_sync = list() | ||
|
r1144 | models = root.findall('models')[0] | ||
for model in models: | ||||
|
r1322 | global_id, exists = GlobalId.from_xml_element(model) | ||
if not exists: | ||||
|
r1386 | print(global_id) | ||
|
r1322 | ids_to_sync.append(global_id) | ||
|
r1386 | print() | ||
|
r1322 | |||
if len(ids_to_sync) > 0: | ||||
|
r1551 | limit = options.get('split_query', len(ids_to_sync)) | ||
for offset in range(0, len(ids_to_sync), limit): | ||||
|
r1563 | xml = GlobalId.objects.generate_request_get(ids_to_sync[offset:offset+limit]) | ||
|
r1551 | # body = urllib.parse.urlencode(data) | ||
h = httplib2.Http() | ||||
response, content = h.request(get_url, method="POST", body=xml) | ||||
|
r1322 | |||
|
r1551 | SyncManager.parse_response_get(content, file_url) | ||
|
r1322 | else: | ||
print('Nothing to get, everything synced') | ||||
|
r1144 | else: | ||
raise Exception('Invalid response status') | ||||