24 lines
859 B
Python
24 lines
859 B
Python
|
from django.core.management.base import BaseCommand, CommandError
|
||
|
|
from events.models.search import Searchable, SearchableSerializer
|
||
|
|
from rest_framework.parsers import JSONParser
|
||
|
|
|
||
|
|
import urllib
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
help = 'Imports searchable data from another node'
|
||
|
|
|
||
|
|
def add_arguments(self, parser):
|
||
|
|
parser.add_argument('url', type=str)
|
||
|
|
|
||
|
|
def handle(self, *args, **options):
|
||
|
|
if 'url' in options:
|
||
|
|
resp = urllib.request.urlopen(options['url'])
|
||
|
|
json_data = JSONParser().parse(resp)
|
||
|
|
serializer = SearchableSerializer(data=json_data, many=True)
|
||
|
|
if serializer.is_valid():
|
||
|
|
serializer.save(federation_node=options['url'])
|
||
|
|
else:
|
||
|
|
print("Serialized data not valid: %s" % serializer.errors)
|
||
|
|
else:
|
||
|
|
print("No URL in options!")
|