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)
108 lines
3.8 KiB
Python
108 lines
3.8 KiB
Python
from django.core.management.base import BaseCommand, CommandError
|
|
|
|
import reversion
|
|
import peeringdb_server.models as pdbm
|
|
import re
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Replace a value in a field across several entities"
|
|
pretend = False
|
|
|
|
# this defines which ref_tag field_name combinations may
|
|
# be targeted by this command, this is a safety measure
|
|
# extend as needed
|
|
valid_targets = {"fac": ["name", "org_id"]}
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
"--commit", action="store_true", help="will commit the fac merge"
|
|
)
|
|
parser.add_argument("--search", help="<ref_tag>.<field_name>:<search_value>")
|
|
parser.add_argument(
|
|
"--replace", help="<field_name>:<search_value>:<replacement_value>"
|
|
)
|
|
|
|
def log(self, msg):
|
|
if not self.commit:
|
|
print("[%s] %s [pretend]" % (self.target, msg))
|
|
else:
|
|
print("[%s] %s" % (self.target, msg))
|
|
|
|
@reversion.create_revision()
|
|
def handle(self, *args, **options):
|
|
|
|
self.commit = options.get("commit", False)
|
|
self.search = options.get("search")
|
|
self.replace = options.get("replace")
|
|
|
|
if not self.search:
|
|
raise CommandError("Specify search parameters using the --search option")
|
|
|
|
if not self.replace:
|
|
raise CommandError(
|
|
"Specify replacement parameters using the --replace option"
|
|
)
|
|
|
|
try:
|
|
search_field, search_value = self.search.split(":")
|
|
ref_tag, search_field = search_field.split(".")
|
|
except:
|
|
raise CommandError(
|
|
"Format for --search: <ref_tag>.<field_name>:<search_value>"
|
|
)
|
|
|
|
try:
|
|
m = re.match("^([^:]+):([^:]+):(.+)$", self.replace)
|
|
replace_field = m.group(1)
|
|
replace_search_value = m.group(2)
|
|
replace_value = m.group(3)
|
|
except:
|
|
raise CommandError(
|
|
"Format for --replace: <field_name>:<search_value>:<replacement_value>"
|
|
)
|
|
|
|
# if replace_field not in self.valid_targets.get(ref_tag,[]):
|
|
# raise CommandError("%s.%s is not a valid target for this script at this point, please add it to the valid_targets map" % (ref_tag, replace_field))
|
|
|
|
self.target = "%s.%s" % (ref_tag, search_field)
|
|
|
|
self.log(
|
|
"Searching for %s where %s matches '%s' ..."
|
|
% (ref_tag, search_field, search_value)
|
|
)
|
|
|
|
q = pdbm.REFTAG_MAP[ref_tag].objects.filter(status="ok")
|
|
c = 0
|
|
|
|
for e in q:
|
|
val = getattr(e, search_field)
|
|
if re.search(search_value, val) != None:
|
|
t_val = getattr(e, replace_field)
|
|
r_val = None
|
|
if replace_search_value == "*":
|
|
r_val = replace_value
|
|
elif isinstance(t_val, str):
|
|
r_val = re.sub(replace_search_value, replace_value, t_val)
|
|
elif type(t_val) == int:
|
|
if t_val == int(replace_search_value or 0):
|
|
r_val = replace_value
|
|
elif type(t_val) == int:
|
|
if t_val == int(replace_search_value or 0):
|
|
r_val = replace_value
|
|
else:
|
|
if t_val == replace_search_value:
|
|
r_val = replace_value
|
|
if r_val is None:
|
|
continue
|
|
self.log(
|
|
"(<%s> id:%s) Changing %s from '%s' to '%s'"
|
|
% (e, e.id, replace_field, t_val, r_val)
|
|
)
|
|
c += 1
|
|
if self.commit:
|
|
setattr(e, replace_field, r_val)
|
|
e.save()
|
|
|
|
self.log("%d objects were changed." % c)
|