mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
ba6f9b6432
* use new peeringdb client (1.0.0) for pdb_load_data sync (#599) * drop django-mobi for lack of py3/dj2 support (#492) remove django-forms-bootstrap for lack of py3/dj2 support (#492) * black formatted * django2.2 and py3 upgrade (#492) * drop ixlans (#21) ui and api changes * drop local_asn (#168) * org search (#193) * phone number validation (#50) * implement help text tooltips (#228) * Mark own ASN as transit-free (#394) * py3 fix for `pdb_migrate_ixlans` command when writing migration report * pdb_migrate_ixlans: properly handle py3 Runtime error if ixlan dict changes during iteration * set rest DEFAULT_SCHEMA_CLASS to coreapi to fix swagger apidocs fix migration 0027 missing from facsimile manifest * fix swagger doc strings * fix tests that were broken from api doc fixes * fix UniqueFieldValidator for netixlan ipaddress validation that broke during django/drf upgrade * fix org merge tool layout issues * travis config * update pipfile and lock * black formatting * update travis dist * beta mode banner (#411) * add beta banner template (#411) * automatically scheduled sync may not always be on, add a flag that lets us reflect that state in the beta banner message clean up beta banner implementation (#411) * add tests for beta banner (#411)
76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
import googlemaps
|
|
import reversion
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from django.conf import settings
|
|
|
|
from peeringdb_server import models
|
|
|
|
API_KEY = settings.GOOGLE_GEOLOC_API_KEY
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Sync latitude and longitude on all geocoding enabled entities"
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
"reftag",
|
|
nargs="?",
|
|
help="can be reftag only such as 'fac' or reftag with id to only sync that specific entity such as 'fac.1'",
|
|
)
|
|
parser.add_argument(
|
|
"--limit",
|
|
type=int,
|
|
default=0,
|
|
help="limit how many rows are synced, useful for testing",
|
|
)
|
|
parser.add_argument(
|
|
"--commit",
|
|
action="store_true",
|
|
help="commit changes, otherwise run in pretend mode",
|
|
)
|
|
|
|
def log(self, msg):
|
|
if not self.commit:
|
|
self.stdout.write("[pretend] {}".format(msg))
|
|
else:
|
|
self.stdout.write(msg)
|
|
|
|
def handle(self, *args, **options):
|
|
self.commit = options.get("commit", False)
|
|
reftag = options.get("reftag")
|
|
limit = options.get("limit")
|
|
if reftag.find(".") > -1:
|
|
reftag, _id = reftag.split(".")
|
|
else:
|
|
_id = 0
|
|
self.gmaps = googlemaps.Client(API_KEY, timeout=5)
|
|
self.sync(reftag, _id, limit=limit)
|
|
|
|
@reversion.create_revision()
|
|
def sync(self, reftag, _id, limit=0):
|
|
model = models.REFTAG_MAP.get(reftag)
|
|
if not model:
|
|
raise ValueError("Unknown reftag: {}".format(reftag))
|
|
if not hasattr(model, "geocode_status"):
|
|
raise TypeError("Can only geosync models containing GeocodeBaseMixin")
|
|
q = model.handleref.undeleted().filter(geocode_status=False)
|
|
if _id:
|
|
q = q.filter(id=_id)
|
|
count = q.count()
|
|
if limit > 0:
|
|
q = q[:limit]
|
|
i = 0
|
|
for entity in q:
|
|
if entity.geocode_status:
|
|
continue
|
|
i += 1
|
|
self.log(
|
|
"Syncing {} [{} {}/{} ID:{}]".format(
|
|
entity.name, reftag, i, count, entity.id
|
|
)
|
|
)
|
|
|
|
if self.commit:
|
|
entity.geocode(self.gmaps)
|