From b1f7b629c7e13a161e93bd04fcc5397febb3d269 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@ribaguifi.com>
Date: Tue, 15 Jun 2021 13:11:49 +0200
Subject: [PATCH] fixing url

---
 .../51439cf24be8_change_trade_action.py       |  6 ++--
 .../resources/tradedocument/models.py         | 31 ++++++-------------
 2 files changed, 14 insertions(+), 23 deletions(-)

diff --git a/ereuse_devicehub/migrations/versions/51439cf24be8_change_trade_action.py b/ereuse_devicehub/migrations/versions/51439cf24be8_change_trade_action.py
index 4f8c87f0..faa88ab8 100644
--- a/ereuse_devicehub/migrations/versions/51439cf24be8_change_trade_action.py
+++ b/ereuse_devicehub/migrations/versions/51439cf24be8_change_trade_action.py
@@ -5,11 +5,12 @@ Revises: eca457d8b2a4
 Create Date: 2021-03-15 17:40:34.410408
 
 """
+import sqlalchemy as sa
+import citext
+import teal
 from alembic import op
 from alembic import context
 from sqlalchemy.dialects import postgresql
-import sqlalchemy as sa
-import citext
 
 
 # revision identifiers, used by Alembic.
@@ -160,6 +161,7 @@ def upgrade():
                     sa.Column(
                         'url',
                         citext.CIText(),
+                        teal.db.URL(),
                         nullable=True,
                         comment='This is the url where resides the document.'
                     ),
diff --git a/ereuse_devicehub/resources/tradedocument/models.py b/ereuse_devicehub/resources/tradedocument/models.py
index 7f15f9b1..6bce51e7 100644
--- a/ereuse_devicehub/resources/tradedocument/models.py
+++ b/ereuse_devicehub/resources/tradedocument/models.py
@@ -1,28 +1,17 @@
-import os
-
-from itertools import chain
 from citext import CIText
-from flask import current_app as app, g
+from flask import g
 
 from sqlalchemy.dialects.postgresql import UUID
 from ereuse_devicehub.db import db
 from ereuse_devicehub.resources.user.models import User
 from sortedcontainers import SortedSet
-from ereuse_devicehub.resources.models import STR_SM_SIZE, Thing, listener_reset_field_updated_in_actual_time
+from ereuse_devicehub.resources.models import Thing
 
-from sqlalchemy import BigInteger, Boolean, Column, Float, ForeignKey, Integer, \
-    Sequence, SmallInteger, Unicode, inspect, text
-from sqlalchemy.ext.declarative import declared_attr
-from sqlalchemy.orm import ColumnProperty, backref, relationship, validates
-from sqlalchemy.util import OrderedSet
-from sqlalchemy_utils import ColorType
-from teal.db import CASCADE_OWN, CASCADE_DEL, POLYMORPHIC_ID, POLYMORPHIC_ON, \
-    check_lower, check_range
-from teal.resource import url_for_resource
+from sqlalchemy import BigInteger, Column, Sequence
+from sqlalchemy.orm import backref
+from teal.db import CASCADE_OWN, URL
 
-from ereuse_devicehub.resources.utils import hashcode
-from ereuse_devicehub.resources.enums import BatteryTechnology, CameraFacing, ComputerChassis, \
-    DataStorageInterface, DisplayTech, PrinterTechnology, RamFormat, RamInterface, Severity, TransferState
+from ereuse_devicehub.resources.enums import Severity
 
 
 _sorted_documents = {
@@ -37,8 +26,8 @@ class TradeDocument(Thing):
     and the action trade need to be confirmed for the both users of the trade.
     This confirmation can be revoked and this revoked need to be ConfirmRevoke for have
     some efect.
-    
-    This documents can be invoices or list of devices or certificates of erasure of 
+
+    This documents can be invoices or list of devices or certificates of erasure of
     one disk.
 
     Like a Devices one document have actions and is possible add or delete of one lot
@@ -68,7 +57,7 @@ class TradeDocument(Thing):
     lot_id = db.Column(UUID(as_uuid=True),
                          db.ForeignKey('lot.id'),
                          nullable=False)
-    lot = db.relationship('Lot', 
+    lot = db.relationship('Lot',
                           backref=backref('documents',
                                           lazy=True,
                                           cascade=CASCADE_OWN,
@@ -79,7 +68,7 @@ class TradeDocument(Thing):
     file_name.comment = """This is the name of the file when user up the document."""
     file_hash = Column(db.CIText())
     file_hash.comment = """This is the hash of the file produced from frontend."""
-    url = Column(db.CIText())
+    url = db.Column(URL())
     url.comment = """This is the url where resides the document."""
 
     __table_args__ = (