97 lines
2.7 KiB
Python
97 lines
2.7 KiB
Python
import csv
|
|
|
|
# import click_spinner
|
|
# import ereuse_utils.cli
|
|
from io import StringIO
|
|
|
|
from ereuse_devicehub.resources.action import models as evs
|
|
from ereuse_devicehub.resources.device.models import Placeholder
|
|
from ereuse_devicehub.resources.documents.device_row import InternalStatsRow
|
|
|
|
# import click
|
|
|
|
|
|
class Report:
|
|
def __init__(self, app) -> None:
|
|
super().__init__()
|
|
self.app = app
|
|
short_help = 'Creates reports devices and users.'
|
|
self.app.cli.command('report', short_help=short_help)(self.run)
|
|
|
|
def run(self):
|
|
stats = InternalStatsView()
|
|
stats.print()
|
|
|
|
|
|
class InternalStatsView:
|
|
def print(self):
|
|
query = evs.Action.query.filter(
|
|
evs.Action.type.in_(
|
|
(
|
|
'Snapshot',
|
|
'Live',
|
|
'Allocate',
|
|
'Deallocate',
|
|
'EraseBasic',
|
|
'EraseSectors',
|
|
)
|
|
)
|
|
)
|
|
return self.generate_post_csv(query)
|
|
|
|
def generate_post_csv(self, query):
|
|
data = StringIO()
|
|
cw = csv.writer(data, delimiter=';', lineterminator="\n", quotechar='"')
|
|
cw.writerow(InternalStatsRow('', "2000-1", [], []).keys())
|
|
|
|
for row in self.get_rows(query):
|
|
cw.writerow(row)
|
|
|
|
return print(data.getvalue())
|
|
|
|
def get_rows(self, query):
|
|
d = {}
|
|
dd = {}
|
|
disks = []
|
|
for ac in query:
|
|
create = '{}-{}'.format(ac.created.year, ac.created.month)
|
|
user = ac.author.email
|
|
|
|
if user not in d:
|
|
d[user] = {}
|
|
dd[user] = {}
|
|
if create not in d[user]:
|
|
d[user][create] = []
|
|
dd[user][create] = None
|
|
d[user][create].append(ac)
|
|
|
|
for user, createds in d.items():
|
|
for create, actions in createds.items():
|
|
r = InternalStatsRow(user, create, actions, disks)
|
|
dd[user][create] = r
|
|
|
|
return self.get_placeholders(dd)
|
|
|
|
def get_placeholders(self, dd):
|
|
|
|
for p in Placeholder.query.all():
|
|
create = '{}-{}'.format(p.created.year, p.created.month)
|
|
user = p.owner.email
|
|
|
|
if user not in dd:
|
|
dd[user] = {}
|
|
|
|
if create not in dd[user]:
|
|
dd[user][create] = None
|
|
|
|
if not dd[user][create]:
|
|
dd[user][create] = InternalStatsRow(user, create, [], [])
|
|
|
|
dd[user][create]['Placeholders'] += 1
|
|
|
|
rows = []
|
|
for user, createds in dd.items():
|
|
for create, row in createds.items():
|
|
rows.append(row.values())
|
|
return rows
|