Model: Change type of AddressOrigin to Enumeration.

pull/19/head
Fabian P. Schmidt 2015-12-02 12:15:44 +01:00
rodzic 6234a4aa11
commit 04137acaf6
2 zmienionych plików z 7 dodań i 3 usunięć

Wyświetl plik

@ -14,7 +14,7 @@ def update_ddb_from_ogn():
logger.info("Update ddb data from ogn.")
app.session.query(Device) \
.filter(Device.address_origin == AddressOrigin.ogn_ddb) \
.filter(AddressOrigin(Device.address_origin) is AddressOrigin.ogn_ddb) \
.delete()
devices = get_ddb()
@ -31,7 +31,7 @@ def update_ddb_from_file():
logger.info("Import ddb data from file.")
app.session.query(Device) \
.filter(Device.address_origin == AddressOrigin.userdefined) \
.filter(AddressOrigin(Device.address_origin) is AddressOrigin.userdefined) \
.delete()
devices = get_ddb('ogn/custom_ddb.txt')

Wyświetl plik

@ -1,4 +1,8 @@
class AddressOrigin:
from enum import Enum, unique
@unique
class AddressOrigin(Enum):
ogn_ddb = 1
flarmnet = 2
userdefined = 3