add user to the name file of json and more readable date

This commit is contained in:
Cayo Puigdefabregas 2020-10-13 15:59:39 +02:00
parent ae98d3154c
commit d3ef6cc65d
2 changed files with 12 additions and 6 deletions

View File

@ -2,7 +2,7 @@
import os import os
import json import json
from time import time from datetime import datetime
from distutils.version import StrictVersion from distutils.version import StrictVersion
from uuid import UUID from uuid import UUID
@ -21,12 +21,18 @@ from ereuse_devicehub.resources.user.exceptions import InsufficientPermission
SUPPORTED_WORKBENCH = StrictVersion('11.0') SUPPORTED_WORKBENCH = StrictVersion('11.0')
def save_json(req_json, tmp_snapshots): def save_json(req_json, tmp_snapshots, user):
""" """
This function allow save a snapshot in json format un a TMP_SNAPSHOTS directory This function allow save a snapshot in json format un a TMP_SNAPSHOTS directory
The file need to be saved with one name format with the stamptime and uuid joins The file need to be saved with one name format with the stamptime and uuid joins
""" """
name_file = "{uuid}_{time}.json".format(uuid=req_json.get('uuid', ''), time=int(time())) uuid = req_json.get('uuid', '')
now = datetime.now()
year = now.year
month = now.month
day = now.day
name_file = f"{uuid}_{user}_{year}-{month}-{day}.json"
path_name = os.path.join(tmp_snapshots, name_file) path_name = os.path.join(tmp_snapshots, name_file)
if not os.path.isdir(tmp_snapshots): if not os.path.isdir(tmp_snapshots):
@ -43,7 +49,7 @@ class ActionView(View):
"""Posts an action.""" """Posts an action."""
json = request.get_json(validate=False) json = request.get_json(validate=False)
tmp_snapshots = app.config['TMP_SNAPSHOTS'] tmp_snapshots = app.config['TMP_SNAPSHOTS']
path_snapshot = save_json(json, tmp_snapshots) path_snapshot = save_json(json, tmp_snapshots, g.user.email)
if not json or 'type' not in json: if not json or 'type' not in json:
raise ValidationError('Resource needs a type.') raise ValidationError('Resource needs a type.')
# todo there should be a way to better get subclassess resource # todo there should be a way to better get subclassess resource

View File

@ -481,11 +481,11 @@ def test_pc_2(user: UserClient):
@pytest.mark.mvp @pytest.mark.mvp
def test_save_snapshot_in_file(app: Devicehub): def test_save_snapshot_in_file(app: Devicehub, user: UserClient):
""" This test check if works the function save_snapshot_in_file """ """ This test check if works the function save_snapshot_in_file """
tmp_snapshots = app.config['TMP_SNAPSHOTS'] tmp_snapshots = app.config['TMP_SNAPSHOTS']
snapshot_no_hid = file('basic.snapshot.nohid') snapshot_no_hid = file('basic.snapshot.nohid')
save_json(snapshot_no_hid, tmp_snapshots) save_json(snapshot_no_hid, tmp_snapshots, user.user['email'])
uuid = snapshot_no_hid['uuid'] uuid = snapshot_no_hid['uuid']
files = [x for x in os.listdir(tmp_snapshots) if uuid in x] files = [x for x in os.listdir(tmp_snapshots) if uuid in x]