add command upload snapshot

ereuse_devicehub/commands/snapshots.py

@@ -0,0 +1,126 @@
+"""This command is used for up one snapshot."""
+
+import copy
+import json
+from io import BytesIO
+
+# from uuid import uuid4
+from pathlib import Path
+
+import click
+from decouple import config
+from flask import Session, g
+
+from ereuse_devicehub.client import UserClient
+from ereuse_devicehub.db import db
+from ereuse_devicehub.resources.action.schemas import Snapshot as SnapshotSchema
+from ereuse_devicehub.resources.action.views.snapshot import SnapshotMixin
+from ereuse_devicehub.resources.device.models import Computer
+from ereuse_devicehub.resources.user.models import User
+
+
+class UploadSnapshots(SnapshotMixin):
+    """
+    Command.
+
+    This command allow upload all snapshots than exist
+    in the directory snapshots_upload.
+    If this snapshot exist replace it.
+
+    """
+
+    def __init__(self, app) -> None:
+        """Init function."""
+        super().__init__()
+        self.app = app
+        self.schema = app.config.get('DB_SCHEMA')
+        self.app.cli.command('snapshot', short_help='Upload snapshots.')(self.run)
+
+    @click.argument('file_snapshot')
+    def run(self, file_snapshot):
+        """Run command."""
+        self.file_snapshot = file_snapshot
+        self.snapshot_json = None
+        self.json_wb = None
+        self.schema = SnapshotSchema()
+
+        with self.app.app_context():
+            self.get_user()
+            self.open_snapshot()
+            self.load_schema()
+            self.build_snapshot()
+            # db.session.commit()
+
+    def get_user(self):
+        """Get datamodel of user."""
+        self.email = config('EMAIL_DEMO')
+        self.password = config('PASSWORD_DEMO')
+        self.user = User.query.filter_by(email=self.email).one()
+        import pdb
+
+        pdb.set_trace()
+        g.user = self.user
+        # if 'dpp' in self.app.blueprints.keys():
+        #    client = UserClient(
+        #        self.app,
+        #        self.email,
+        #        self.password,
+        #        response_wrapper=self.app.response_class
+        #    )
+        #    client.login_web(self.email, self.password)
+
+        from flask.testing import FlaskClient
+        from flask_wtf.csrf import generate_csrf
+
+        client = FlaskClient(self.app, use_cookies=True)
+
+        body, status, headers = client.get('/login/')
+
+        data = {
+            'email': self.email,
+            'password': self.password,
+            'remember': False,
+            'csrf_token': generate_csrf(),
+        }
+        self.client.post('/login/', data=data, follow_redirects=True)
+
+    def open_snapshot(self):
+        """Open snapshot file."""
+        with Path(__file__).parent.joinpath('snapshot_files').joinpath(
+            self.file_snapshot
+        ).open() as file_snapshot:
+            self.json_wb = json.loads(file_snapshot.read())
+            b_snapshot = bytes(json.dumps(snapshot), 'utf-8')
+            self.file_snap = (BytesIO(b_snapshot), self.file_snapshot)
+
+    def load_schema(self):
+        """Load schema for check snapshot."""
+        if not self.json_wb:
+            return
+        self.snapshot_json = self.schema.load(self.json_wb)
+
+    def build_snapshot(self):
+        """Build the devices of snapshot."""
+        import pdb
+
+        pdb.set_trace()
+        if not self.snapshot_json:
+            return
+        response = self.build(self.snapshot_json)
+        if isinstance(response.device, Computer):
+            response.device.user_trusts = True
+            db.session.add(response)
+
+    uri = '/inventory/upload-snapshot/'
+    snapshot = conftest.yaml2json(file_name.split(".json")[0])
+    b_snapshot = bytes(json.dumps(snapshot), 'utf-8')
+    file_snap = (BytesIO(b_snapshot), file_name)
+    user.get(uri)
+
+    data = {
+        'snapshot': file_snap,
+        'csrf_token': generate_csrf(),
+    }
+    user.post(uri, data=data, content_type="multipart/form-data")
+
+    return Snapshot.query.one()

ereuse_devicehub/devicehub.py

@@ -14,6 +14,7 @@ from ereuse_devicehub.auth import Auth
 from ereuse_devicehub.client import Client, UserClient
 from ereuse_devicehub.commands.adduser import AddUser
 from ereuse_devicehub.commands.initdatas import InitDatas
+from ereuse_devicehub.commands.snapshots import UploadSnapshots
 
 # from ereuse_devicehub.commands.reports import Report
 from ereuse_devicehub.commands.users import GetToken
@@ -102,6 +103,7 @@ class Devicehub(Teal):
         self.get_token = GetToken(self)
         self.initdata = InitDatas(self)
         self.adduser = AddUser(self)
+        self.uploadsnapshots = UploadSnapshots(self)
 
         if GetMembers:
             self.get_members = GetMembers(self)