34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
|
import whoisdb.models
|
||
|
|
|
||
|
|
def _addFields(fields, obj, fieldNames):
|
||
|
|
for fieldName in fieldNames:
|
||
|
|
fields.append((fieldName.capitalize(), getattr(obj, fieldName)))
|
||
|
|
|
||
|
|
|
||
|
|
def getWhoisObjectFields(obj, owner):
|
||
|
|
fields = []
|
||
|
|
|
||
|
|
if getattr(obj, "handle"):
|
||
|
|
_addFields(fields, obj, ["handle"])
|
||
|
|
|
||
|
|
c = type(obj)
|
||
|
|
if c == whoisdb.models.Maintainer:
|
||
|
|
_addFields(fields, obj, ["admin_c"])
|
||
|
|
if owner:
|
||
|
|
_addFields(fields, obj, ["auth"])
|
||
|
|
elif c == whoisdb.models.Contact:
|
||
|
|
_addFields(fields, obj, ["name", "mnt_by"])
|
||
|
|
elif c == whoisdb.models.ASBlock:
|
||
|
|
_addFields(fields, obj, ["name"])
|
||
|
|
fields.append(("AS Range", "%s - %s" % (obj.asBegin, obj.asEnd)))
|
||
|
|
_addFields(fields, obj, ["description", "mnt_by", "mnt_lower", "admin_c"])
|
||
|
|
elif c == whoisdb.models.ASNumber:
|
||
|
|
_addFields(fields, obj, ["name", "number", "description", "mnt_by", "mnt_lower", "admin_c"])
|
||
|
|
elif c == whoisdb.models.InetNum:
|
||
|
|
_addFields(fields, obj, ["name"])
|
||
|
|
fields.append(("Address CIDR", obj.prefix()))
|
||
|
|
_addFields(fields, obj, ["description", "mnt_by", "mnt_lower", "admin_c"])
|
||
|
|
|
||
|
|
return fields
|
||
|
|
|