mirror of
https://github.com/Eledio/superfaktura-client.git
synced 2025-10-31 16:11:20 +01:00
100 lines
2.7 KiB
Python
100 lines
2.7 KiB
Python
import dataclasses
|
|
import json
|
|
from pprint import pprint
|
|
from typing import Optional
|
|
|
|
from superfaktura.superfaktura_api import SuperFakturaAPI
|
|
|
|
|
|
class ClientException(Exception):
|
|
pass
|
|
|
|
|
|
@dataclasses.dataclass
|
|
class ClientContactModel:
|
|
name: str
|
|
address: Optional[str] = None
|
|
bank_account: Optional[str] = None
|
|
bank_code: Optional[str] = None
|
|
city: Optional[str] = None
|
|
comment: Optional[str] = None
|
|
country: Optional[str] = None
|
|
country_id: Optional[int] = None
|
|
currency: Optional[str] = None
|
|
default_variable: Optional[str] = None
|
|
delivery_address: Optional[str] = None
|
|
delivery_city: Optional[str] = None
|
|
delivery_country: Optional[str] = None
|
|
delivery_country_id: Optional[int] = None
|
|
delivery_name: Optional[str] = None
|
|
delivery_phone: Optional[str] = None
|
|
delivery_zip: Optional[str] = None
|
|
dic: Optional[str] = None
|
|
discount: Optional[float] = None
|
|
due_date: Optional[int] = None
|
|
email: Optional[str] = None
|
|
fax: Optional[str] = None
|
|
iban: Optional[str] = None
|
|
ic_dph: Optional[str] = None
|
|
ico: Optional[str] = None
|
|
match_address: Optional[int] = None
|
|
phone: Optional[str] = None
|
|
swift: Optional[str] = None
|
|
tags: Optional[str] = None
|
|
uuid: Optional[str] = None
|
|
zip: Optional[str] = None
|
|
update: Optional[bool] = None
|
|
|
|
id: Optional[int] = None
|
|
|
|
def as_dict(self) -> dict:
|
|
data = dataclasses.asdict(self)
|
|
for key in list(data.keys()):
|
|
if data[key] is None:
|
|
del data[key]
|
|
return data
|
|
|
|
@staticmethod
|
|
def from_dict(data: dict) -> "ClientContactModel":
|
|
return ClientContactModel(
|
|
**{
|
|
k: v
|
|
for k, v in data.items()
|
|
if k in ClientContactModel.__dataclass_fields__
|
|
}
|
|
)
|
|
|
|
|
|
class ClientContact(SuperFakturaAPI):
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def add_contact(self, contact: ClientContactModel) -> bool:
|
|
url = "clients/create"
|
|
data = {"Client": contact.as_dict()}
|
|
resp = self.post(endpoint=url, data=json.dumps(data))
|
|
if resp["error_message"] == "Client created":
|
|
return True
|
|
return False
|
|
|
|
def list(self) -> dict:
|
|
url = "clients/index.json"
|
|
return self.get(endpoint=url)
|
|
|
|
def get_client(self, client_id: int) -> ClientContactModel:
|
|
url = f"clients/view/{client_id}"
|
|
data = self.get(endpoint=url)
|
|
if "Client" not in data:
|
|
raise ClientException("Client not found")
|
|
data = data["Client"]
|
|
return ClientContactModel.from_dict(data)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
client = ClientContact()
|
|
resp = client.list()
|
|
|
|
pprint(resp)
|
|
|
|
pprint(client.get_client(40019))
|