47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
from PySide6.QtCore import QAbstractListModel, Qt, Slot, QModelIndex
|
|
from .AddressDAO import AddressDAO
|
|
from ..PyqcrmDataRoles import PyqcrmDataRoles
|
|
|
|
class AddressModel(QAbstractListModel):
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.__address_data = AddressDAO().getAddressData()
|
|
|
|
def rowCount(self, parent = QModelIndex()):
|
|
return len(self.__address_data)
|
|
|
|
def data(self, index, role = Qt.DisplayRole):
|
|
row = index.row()
|
|
if role == Qt.DisplayRole:
|
|
data = self.__address_data[row][5]
|
|
return data
|
|
elif role == PyqcrmDataRoles.CITY_ROLE:
|
|
data = self.__address_data[row][4]
|
|
return data
|
|
elif role == PyqcrmDataRoles.COUNTRY_ROLE:
|
|
data = self.__address_data[row][3]
|
|
return data
|
|
return None
|
|
|
|
def roleNames(self):
|
|
return {
|
|
Qt.DisplayRole: b"display",
|
|
PyqcrmDataRoles.CITY_ROLE: b"city",
|
|
PyqcrmDataRoles.COUNTRY_ROLE: b"country"
|
|
}
|
|
|
|
def setData(self):
|
|
pass
|
|
|
|
@Slot(bool, str)
|
|
def getAddresses(self, all, zipcode):
|
|
data = AddressDAO().getAddressData(all, zipcode)
|
|
return data
|
|
|
|
|
|
|
|
|
|
|
|
|