From ab3510b67bb4a2f953d053f3e6f33fdeadd0cd21 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@ribaguifi.com>
Date: Tue, 25 May 2021 12:57:21 +0200
Subject: [PATCH] fixed bug

---
 .../resources/documents/documents.py            | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/ereuse_devicehub/resources/documents/documents.py b/ereuse_devicehub/resources/documents/documents.py
index 3b72d85d..37b78397 100644
--- a/ereuse_devicehub/resources/documents/documents.py
+++ b/ereuse_devicehub/resources/documents/documents.py
@@ -1,6 +1,7 @@
 import csv
 import enum
 import uuid
+import time
 import datetime
 import pathlib
 from collections import OrderedDict
@@ -22,6 +23,7 @@ from teal.resource import Resource, View
 from ereuse_devicehub import auth
 from ereuse_devicehub.db import db
 from ereuse_devicehub.resources.enums import SessionType
+from ereuse_devicehub.resources.user.models import Session
 from ereuse_devicehub.resources.action import models as evs
 from ereuse_devicehub.resources.device import models as devs
 from ereuse_devicehub.resources.deliverynote.models import Deliverynote
@@ -336,7 +338,20 @@ class WbConfDocumentView(DeviceView):
         return output
 
     def get_token(self):
-        tk = [s.token for s in g.user.sessions if s.type == SessionType.Internal][0]
+        if not g.user.sessions:
+            ses = Session(user=g.user)
+            db.session.add(ses)
+            db.session.commit()
+
+        tk = ''
+        now = time.time()
+        for s in g.user.sessions:
+            if s.type == SessionType.Internal and (s.expired == 0 or s.expired > now):
+                tk = s.token
+                break
+
+        assert tk != ''
+
         token = auth.Auth.encode(tk)
         return token