Merge pull request #11 from Ivanov-Anton/package-counters-entity

add package-counters entity for Admin API client

Author: dmitry-sinina

examples/package_counters.py

@@ -0,0 +1,30 @@
+from yeti_switch_api.orm import PackageCounter, OrmClient
+
+API_ROOT = "http://127.0.0.1:3000/api/rest/admin"
+USERNAME = "admin"
+PASSWORD = "111111"
+
+config = {
+    "API_ROOT": API_ROOT,
+    "AUTH_CREDS": {"login": USERNAME, "password": PASSWORD}
+}
+
+# Initialize ORM client (registers models)
+OrmClient(config)
+
+# Fetch list of package counters
+counters = PackageCounter.get_list()
+print(f"Total counters: {len(counters)}")
+for counter in counters:
+    print(f"ID: {counter.id}, Prefix: {counter.prefix}, Duration: {counter.duration}")
+
+# Fetch first counter by ID (if any exist)
+if counters:
+    first_id = counters[0].id
+    counter = PackageCounter.find_by_id(first_id)
+    print("\nFirst counter details:")
+    print(f"ID: {counter.id}")
+    print(f"Attributes: {counter.raw_object.attributes}")
+    print(f"Relationships: {counter.raw_object.relationships}")
+else:
+    print("No package counters found.")

yeti_switch_api/orm/__init__.py

@@ -25,3 +25,4 @@
 from .network import Network  # noqa: F401
 from .network_type import NetworkType  # noqa: F401
 from .timezone import Timezone  # noqa: F401
+from .package_counter import PackageCounter  # noqa: F401

yeti_switch_api/orm/orm_client.py

@@ -28,6 +28,7 @@
 from .network import Network
 from .network_type import NetworkType
 from .timezone import Timezone
+from .package_counter import PackageCounter
 
 
 class OrmClient:
@@ -67,6 +68,7 @@ def __register_models(cls):
         cls.__register_model(Network)
         cls.__register_model(NetworkType)
         cls.__register_model(Timezone)
+        cls.__register_model(PackageCounter)
 
     @classmethod
     def __register_model(cls, model_class):

yeti_switch_api/orm/package_counter.py

@@ -0,0 +1,14 @@
+from .base_model import BaseModel, AttributeField, RelationField
+
+class PackageCounter(BaseModel):
+    class Meta:
+        path = "package-counters"
+        type = "package-counters"
+
+    duration = AttributeField("duration")
+    exclude = AttributeField("exclude")
+    prefix = AttributeField("prefix")
+    service_id = AttributeField("service-id")
+
+    account = RelationField("account")
+    service = RelationField("service")