1
0
mirror of https://github.com/peeringdb/peeringdb.git synced 2024-05-11 05:55:09 +00:00
Files
peeringdb-peeringdb/peeringdb_server/management/commands/pdb_whois.py
Matt Griswold ba6f9b6432 Qu1003 (#621)
* 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)
2020-01-08 13:29:58 -06:00

61 lines
1.9 KiB
Python

import logging
from ._db_command import CommandError, DBCommand
from django.contrib.auth.models import AnonymousUser
from peeringdb.whois import WhoisFormat
from peeringdb_server import models
from peeringdb_server import serializers
from django_handleref import util
class Command(DBCommand):
args = "<customer id>"
help = "command line whois"
def add_arguments(self, parser):
parser.add_argument("ref", nargs="?", type=str)
def handle(self, *args, **options):
log = logging.getLogger("pdb.script.whois")
# FIXME - ignore multiple args for now
args = options.get("ref")
try:
(ref_tag, pk) = util.split_ref(args)
except ValueError:
log.error("Unknown query type '%s'" % (args))
return
# TODO
raise CommandError("unk query")
model = None
# TODO needs the client whois typ resolver in a better place for reuse
# probably easiest to just map reftag to pk name
if ref_tag in models.REFTAG_MAP:
model = models.REFTAG_MAP[ref_tag]
Serializer = serializers.REFTAG_MAP[ref_tag]
obj = Serializer.prefetch_related(model.objects, None, depth=2).get(pk=pk)
elif ref_tag == "as":
model = models.REFTAG_MAP["net"]
Serializer = serializers.REFTAG_MAP["net"]
obj = Serializer.prefetch_related(model.objects, None, depth=2).get(asn=pk)
# data = cls(obj).data
# TODO doesn't work on client
# elif ref_tag == 'ixnets':
if not model:
msg = "Unknown ref tag: {}".format(ref_tag)
log.error("Unknown ref tag: %s" % ref_tag)
raise ValueError(msg)
data = Serializer(obj, context={"user": AnonymousUser()}).data
fmt = WhoisFormat()
fmt.print(obj._handleref.tag, data)