def get_count_stats():
day_ago = datetime.datetime.now() - datetime.timedelta(hours=24)
week_ago = datetime.datetime.now() - datetime.timedelta(days=7)
month_ago = datetime.datetime.now() - datetime.timedelta(days=30)
incoming = {
"today": ReceiveStatistic.select().where(ReceiveStatistic.created_at >= day_ago).count(),
"week": ReceiveStatistic.select().where(ReceiveStatistic.created_at >= week_ago).count(),
"month": ReceiveStatistic.select().where(ReceiveStatistic.created_at >= month_ago).count(),
"all_time": ReceiveStatistic.select().count(),
}
outgoing = {
"today": WorkerReceiveStatistic.select().where(WorkerReceiveStatistic.created_at >= day_ago).count(),
"week": WorkerReceiveStatistic.select().where(WorkerReceiveStatistic.created_at >= week_ago).count(),
"month": WorkerReceiveStatistic.select().where(WorkerReceiveStatistic.created_at >= month_ago).count(),
"all_time": WorkerReceiveStatistic.select().count(),
}
distinct_nodes = {
"all": Node.select().count(),
"https": Node.select().where(Node.https==True).count(),
}
processing = {
"workers": get_worker_count(),
"queue_jobs": len(public_queue),
}
return incoming, outgoing, distinct_nodes, processing
def log_worker_receive_statistics(protocol, entities, sent_amount,
sent_success):
"""Add a WorkerReceiveStatistic entry to the database."""
WorkerReceiveStatistic.create(protocol=protocol,
entities=entities,
sent_amount=sent_amount,
sent_success=sent_success)
def test_worker_receive_statistics_logged(self):
log_worker_receive_statistics("diaspora", 3, 2, 1)
assert WorkerReceiveStatistic.select().count() == 1
statistic = WorkerReceiveStatistic.select().first()
assert statistic.protocol == "diaspora"
assert statistic.entities == 3
assert statistic.sent_amount == 2
assert statistic.sent_success == 1
def get_count_stats():
day_ago = datetime.datetime.now() - datetime.timedelta(hours=24)
week_ago = datetime.datetime.now() - datetime.timedelta(days=7)
month_ago = datetime.datetime.now() - datetime.timedelta(days=30)
incoming = {
"today":
ReceiveStatistic.select().where(
ReceiveStatistic.created_at >= day_ago).count(),
"week":
ReceiveStatistic.select().where(
ReceiveStatistic.created_at >= week_ago).count(),
"month":
ReceiveStatistic.select().where(
ReceiveStatistic.created_at >= month_ago).count(),
"all_time":
ReceiveStatistic.select().count(),
}
outgoing = {
"today":
WorkerReceiveStatistic.select().where(
WorkerReceiveStatistic.created_at >= day_ago).count(),
"week":
WorkerReceiveStatistic.select().where(
WorkerReceiveStatistic.created_at >= week_ago).count(),
"month":
WorkerReceiveStatistic.select().where(
WorkerReceiveStatistic.created_at >= month_ago).count(),
"all_time":
WorkerReceiveStatistic.select().count(),
}
distinct_nodes = {
"all": Node.select().count(),
}
processing = {
"workers": get_worker_count(),
"queue_jobs": len(public_queue),
}
profiles = {
"all": Profile.select().count(),
}
return incoming, outgoing, distinct_nodes, processing, profiles
def log_worker_receive_statistics(protocol, entities, sent_amount, sent_success):
"""Add a WorkerReceiveStatistic entry to the database."""
WorkerReceiveStatistic.create(
protocol=protocol, entities=entities, sent_amount=sent_amount, sent_success=sent_success
)
