mirror of
https://github.com/nikdoof/vapemap.git
synced 2025-12-14 14:52:16 +00:00
52 lines
2.4 KiB
Python
52 lines
2.4 KiB
Python
import csv
|
|
import requests
|
|
from StringIO import StringIO
|
|
from django.db import transaction
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
from stores.models import Country, County, Address, Store, Link, LinkType
|
|
from django.contrib.contenttypes.models import ContentType
|
|
|
|
|
|
class Command(BaseCommand):
|
|
args = '<file or url>'
|
|
help = 'Import a list of stores from CSV'
|
|
|
|
def handle(self, *args, **options):
|
|
file = args[0]
|
|
if file.startswith('http'):
|
|
self.stdout.write("Downloading %s\n" % file)
|
|
f = StringIO(requests.get(file).text)
|
|
else:
|
|
self.stdout.write("Opening file %s\n" % file)
|
|
f = open(file, 'r')
|
|
|
|
for row in csv.reader(f):
|
|
row = [x.strip() for x in row]
|
|
name, addr1, addr2, addr3, city, county, country, postcode, email, website, phone, y, x, twitter, facebook, facebook2 = row
|
|
self.stdout.write("Importing %s... " % name)
|
|
try:
|
|
obj = Store.objects.get(name=name)
|
|
except Store.DoesNotExist:
|
|
pass
|
|
else:
|
|
if obj.address.postcode == postcode:
|
|
self.stdout.write("Store by that name already exists, skipping.\n")
|
|
continue
|
|
with transaction.commit_on_success():
|
|
country, created = Country.objects.get_or_create(name=country)
|
|
county, created = County.objects.get_or_create(name=county, country=country)
|
|
addr = Address(name=name, address1=addr1, address2=addr2, address3=addr3, city=city, county=county, country=country, postcode=postcode, geo_latitude=y, geo_longitude=x)
|
|
addr.save(no_lookup=True)
|
|
store = Store(name=name, address=addr, website=website, email=email, phone=phone)
|
|
if website:
|
|
store.store_type = Store.STORE_TYPE_BOTH
|
|
store.save()
|
|
if twitter:
|
|
Link(object_id=store.pk, object_type=ContentType.objects.get_for_model(store.__class__), account_type=LinkType.objects.get(name='Twitter'), account_name=twitter.split('/')[-1]).save()
|
|
#if facebook:
|
|
# Link(object_id=store.pk, object_type=ContentType.objects.get_for_model(store.__class__), account_type=LinkType.objects.get(name='Facebook'), account_name=facebook).save()
|
|
self.stdout.write("Done\n")
|
|
|
|
f.close()
|
|
self.stdout.write("Done\n")
|