mirror of
https://github.com/peeringdb/peeringdb.git
synced 2024-05-11 05:55:09 +00:00
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
from __future__ import absolute_import, division, print_function
|
|
|
|
import datetime
|
|
import json
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from django.core.management.base import BaseCommand
|
|
from django.contrib.contenttypes.models import ContentType
|
|
import reversion
|
|
from reversion.models import Version, Revision
|
|
|
|
from peeringdb_server.models import REFTAG_MAP, UTC
|
|
|
|
|
|
class Command(BaseCommand):
|
|
"""
|
|
Posts stat breakdown for any given date, if not date is supplied
|
|
today will be used
|
|
"""
|
|
|
|
tags = ["fac", "ix", "net", "org"]
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument("--date", action="store", default=None,
|
|
help="generate stats for this date")
|
|
parser.add_argument("--format", action="store", default="text",
|
|
help="output format to use")
|
|
|
|
def status_at_date(self, obj, dt):
|
|
versions = Version.objects.get_for_object(obj)
|
|
version = versions.filter(revision__date_created__lte=dt).order_by(
|
|
"-revision__date_created").first()
|
|
if version:
|
|
return version.field_dict["status"]
|
|
else:
|
|
return obj.status
|
|
|
|
def handle(self, *args, **options):
|
|
date = options.get('date', None)
|
|
if date:
|
|
dt = datetime.datetime.strptime(date, "%Y%m%d")
|
|
else:
|
|
dt = datetime.datetime.now()
|
|
|
|
dt = dt.replace(hour=23, minute=23, second=59, tzinfo=UTC())
|
|
date = dt.replace(tzinfo=None).strftime("%Y-%m-%d")
|
|
stats = {"users": 0}
|
|
|
|
for tag in self.tags:
|
|
model = REFTAG_MAP[tag]
|
|
stats[tag] = 0
|
|
for obj in model.objects.filter(created__lte=dt):
|
|
if self.status_at_date(obj, dt) == "ok":
|
|
stats[tag] += 1
|
|
|
|
for user in get_user_model().objects.filter(created__lte=dt):
|
|
if user.is_verified:
|
|
stats["users"] += 1
|
|
|
|
codec = options.get("format")
|
|
if codec == "text":
|
|
print(date)
|
|
print("-------------")
|
|
for each in stats.keys():
|
|
print("{}: {}".format(each, stats[each]))
|
|
|
|
elif codec == "json":
|
|
print(json.dumps({date: stats}))
|
|
|
|
else:
|
|
raise Exception("unknown format {}".format(codec))
|