def fetchdata() -> Any:
if request.method == 'GET':
keys = request.values.getlist('keys[]')
parse = request.values.get('parse', '1')
now = datetime.utcnow()
if 'from' in request.values:
from_date = parseTimestamp(request.values['from'])
else:
from_date = now - timedelta(days=41)
if 'to' in request.values:
to_date = parseTimestamp(request.values['to'])
else:
to_date = now
if getattr(settings, 'OVERRIDE_START', None):
from_date = settings.OVERRIDE_START
if getattr(settings, 'OVERRIDE_END', None):
to_date = settings.OVERRIDE_END
if keys and keys[0] == "all":
keys = None
parser = Parser(from_date, to_date, modules=keys)
if parse == '1':
parser.parse()
data: Dict[str, Dict] = {}
for key in settings.ENABLED_MODULES:
if not keys or key in keys:
data[key] = {}
data[key]['entries'] = [
entry.to_dict() for entry in Entry.query.filter(
Entry.module == key, Entry.start_time >= from_date,
Entry.start_time <= to_date)
]
data[key]['projects'] = [
project.to_dict()
for project in Project.query.filter(Project.module == key)
]
data[key]['issues'] = [
issue.to_dict()
for issue in Issue.query.filter(Issue.module == key)
]
data[key]['capabilities'] = parser.modules[key].capabilities
data[key]['color'] = settings.UI_SETTINGS.get(key, {})
return json.dumps(data, default=json_serial)
return None
def listmodules() -> Any:
if request.method == 'GET':
data = {}
parser = Parser(None, None)
for module_name in settings.ENABLED_MODULES:
data[module_name] = {
'color': settings.UI_SETTINGS.get(module_name, {}),
'capabilities': parser.modules[module_name].capabilities,
}
return json.dumps(data, default=json_serial)
return None
def updateentry() -> Any:
if request.method == 'POST':
data = request.get_json()
module = data.get('module')
entry_id = _str(data.get('entry_id', None))
project = data.get('project_id', None)
if project == "null":
project = None
project_to_group = {
project.pid: project.group
for project in Project.query.all()
}
new_entry = Entry(start_time=parseTimestamp(data['start_time']),
end_time=parseTimestamp(data.get('end_time', None)),
id=entry_id,
issue=data.get('issue_id', None),
project=project,
title=data.get('title', ''),
text=data.get('text', ""),
extra_data=data.get('extra_data', {}),
group=project_to_group.get(str(project), None))
issue = None
if new_entry.issue:
issue = Issue.query.filter(Issue.uuid == new_entry.issue).first()
parser = Parser(None, None)
if not entry_id:
# Check that create is allowed
assert isinstance(parser.modules[module], AddEntryMixin)
new_entry = parser.modules[module].create_entry( # type: ignore
new_entry=new_entry, issue=issue)
else:
# Check that update is allowed
assert isinstance(parser.modules[module], UpdateEntryMixin)
new_entry = parser.modules[module].update_entry( # type: ignore
entry_id=new_entry.id,
new_entry=new_entry,
issue=issue)
data = "error"
if new_entry:
Entry.query.filter(Entry.id == new_entry.id).delete()
new_entry.module = module
db.session.merge(new_entry)
db.session.commit()
data = new_entry.to_dict()
return json.dumps(data, default=json_serial)
return None
def deleteentry() -> Any:
if request.method == 'POST':
data = request.get_json()
module = data.get('module')
entry_id = data.get('entry_id')
parser = Parser(None, None)
# Check that delete is allowed
assert isinstance(parser.modules[module], AddEntryMixin)
ret = parser.modules[module].delete_entry( # type: ignore
entry_id=entry_id)
Entry.query.filter(Entry.id == entry_id).delete()
db.session.commit()
return json.dumps(ret, default=json_serial)
return None