try execute create_code

This commit is contained in:
Cayo Puigdefabregas 2021-03-04 10:19:31 +01:00
parent 269550270f
commit 41cadcc0b3
2 changed files with 16 additions and 8 deletions

View File

@ -5,10 +5,11 @@ Revises: eca457d8b2a4
Create Date: 2021-03-03 10:39:19.331027 Create Date: 2021-03-03 10:39:19.331027
""" """
import citext
import sqlalchemy as sa
from alembic import op from alembic import op
from alembic import context from alembic import context
import sqlalchemy as sa from ereuse_devicehub.resources.device.utils import Hashids
import citext
# revision identifiers, used by Alembic. # revision identifiers, used by Alembic.
@ -25,8 +26,13 @@ def get_inv():
return INV return INV
def create_code(context):
_id = context.get_current_parameters()['id']
return Hashids(_id)
def upgrade(): def upgrade():
op.add_column('device', sa.Column('code', citext.CIText(), op.add_column('device', sa.Column('code', citext.CIText(),
default=create_code,
nullable=True), schema=f'{get_inv()}') nullable=True), schema=f'{get_inv()}')

View File

@ -1029,10 +1029,12 @@ class Manufacturer(db.Model):
listener_reset_field_updated_in_actual_time(Device) listener_reset_field_updated_in_actual_time(Device)
def create_code(mapper, connection, thing_obj): # def create_code(mapper, connection, thing_obj):
""" This function create the code for every device.""" # """ This function create the code for every device."""
thing_obj.code = Hashids(thing_obj.id) # thing_obj.code = Hashids(thing_obj.id)
def listener_reset_field_updated_in_actual_time(thing_obj): # def listener_create_code(thing_obj):
""" This function launch a event than listen like a signal when some device is create.""" # """ This function launch a event than listen like a signal when some device is create."""
event.listen(thing_obj, 'after_create', create_code, propagate=True) # event.listen(thing_obj, 'before_create', create_code)
# listener_create_code(Device)