flowtask 5.8.4__cp312-cp312-manylinux2014_x86_64.manylinux_2_17_x86_64.whl

flowtask/components/NetworkNinja/models/project.py

@@ -0,0 +1,44 @@
+from typing import Optional
+from datetime import datetime
+from datamodel import Field
+from .abstract import AbstractPayload
+from .organization import Organization
+from .client import Client
+
+
+class Project(AbstractPayload):
+    """
+    NN Projects related to a Client.
+
+    Example payload:
+    "payload": {
+        "project_id": 595,
+        "project_name": "Test Program",
+        "is_active": true,
+        "start_timestamp": "2025-02-01T00:00:00-06:00",
+        "end_timestamp": "2025-02-28T00:00:00-06:00",
+        "inserted_at": "2025-01-30T19:19:40-06:00",
+        "updated_at": "2025-01-30T19:19:40-06:00",
+        "description": "",
+        "orgid": 69,
+        "client_id": 56,
+        "client_name": "EPSON"
+    }
+    """
+    project_id: int = Field(primary_key=True, required=True)
+    project_name: str
+    is_active: bool = Field(default=True)
+    start_timestamp: datetime
+    end_timestamp: datetime
+    inserted_at: datetime
+    updated_at: datetime
+    description: str
+    orgid: Optional[Organization]
+    client_id: Client
+    client_name: str
+
+    class Meta:
+        strict = True
+        as_objects = True
+        name = 'projects'
+        schema: str = 'networkninja'

flowtask/components/NetworkNinja/models/region.py

@@ -0,0 +1,28 @@
+from typing import Any
+from datamodel import BaseModel, Field
+
+class Region(BaseModel):
+    region_id: str = Field(required=False, alias='id')
+    region_name: str = Field(required=False, alias='name')
+    is_active: bool = Field(required=False, alias='active')
+    client_id: int
+    orgid: int
+
+    class Meta:
+        name: str = 'regions'
+        schema: str = 'networkninja'
+        strict: bool = True
+
+# def create_region(
+#     name: str,
+#     value: Any,
+#     target_type: Any,
+#     parent_data: BaseModel
+# ) -> Region:
+#     client_id = parent_data.get('client_id', None) if parent_data else None
+#     print('Parent Data: ', name, value, target_type, client_id)
+#     # return target_type(**args)
+#     return Region(region_id='1', region_name='Test Region')
+
+# # TODO: To be implemented in Datamodel.
+# BaseModel.register_parser(Region, create_region, 'region_name')

flowtask/components/NetworkNinja/models/store.py

@@ -0,0 +1,203 @@
+from typing import List, Optional, Union, Dict, Any
+from datetime import datetime
+from datamodel import BaseModel, Field
+from slugify import slugify
+from .abstract import AbstractPayload
+from .organization import Organization
+from .client import Client
+from .region import Region
+from .market import Market
+from .district import District
+
+
+class StoreGeography(AbstractPayload):
+    """
+    Store Geography Model.
+
+    Represents a store's geographical information.
+
+    Example:
+        {
+            "geoid": 479,
+            "region": "Assembly - Region",
+            "district": "Assembly - District",
+            "market": "136",
+            "company_id": 61,
+            "orgid": 71,
+            "client_id": 61,
+            "client_name": "ASSEMBLY"
+        }
+    """
+    geoid: int = Field(primary_key=True, required=True)
+    region: str
+    region_id: int
+    district_id: int
+    district: str
+    market_id: int
+    market: str
+    company_id: int
+    orgid: Organization
+    client_id: Client
+    client_name: str
+
+    class Meta:
+        strict = True
+        as_objects = True
+        name = 'stores_geographies'
+        schema: str = 'networkninja'
+
+    def __post_init__(self):
+        super().__post_init__()
+        self.client_id.client_name = self.client_name
+
+class StoreType(AbstractPayload):
+    """
+    Store Type Model.
+
+    Represents a store type in the system.
+
+    Example:
+        {
+            "store_type_id": 1,
+            "store_type_name": "Retail",
+            "store_type_description": "Retail Store"
+        }
+    """
+    store_type_id: int = Field(primary_key=True, required=True)
+    store_type: str = Field(alias="store_type_name")
+    description: str
+    client_id: Client
+    client_name: str
+    status: bool = Field(default=True)
+
+    class Meta:
+        strict = True
+        as_objects = True
+        name = 'stores_types'
+        schema: str = 'networkninja'
+
+
+class CustomStoreField(BaseModel):
+    """
+    Custom Field Model for Store.
+
+    Represents a custom field for a store.
+
+    Example:
+        {
+        "custom_id": 33,
+        "custom_name": "Store Name",
+        "custom_value": "Best Buy 4350",
+        "custom_orgid": null,
+        "custom_client_id": 1
+    }
+    """
+    store_id: int
+    custom_id: int = Field(primary_key=True, required=True)
+    name: str = Field(alias="custom_name")
+    column_name: Union[str, int]
+    value: Union[str, None] = Field(alias="custom_value")
+    obj_type: str = Field(alias="custom_type", default="Text")
+    orgid: int = Field(alias="custom_orgid")
+    client_id: str = Field(alias="custom_client_id")
+
+    class Meta:
+        name = 'stores_attributes'
+        schema: str = 'networkninja'
+
+    def __post_init__(self):
+        self.column_name = slugify(self.name, separator="_")
+        return super().__post_init__()
+
+    def get_field(self):
+        return {
+            self.column_name: self.value
+        }
+
+def default_timezone(*args, **kwargs):
+    return "America/New_York"
+
+
+class Store(AbstractPayload):
+    """
+    Store Model.
+
+    Represents a store in the system.
+
+    Example:
+        {
+            "store_name": "KILMARNOCK-4350",
+            "store_address": "200 Old Fair Grounds Way",
+            "city": "Kilmarnock",
+            "zipcode": "22482",
+            "phone_number": "804-435-6149",
+        }
+    """
+    store_number: int = Field(primary_key=True, required=True, alias='store_id')
+    store_id: Optional[str]
+    store_name: str
+    store_address: str
+    city: str
+    zipcode: str
+    phone_number: Optional[str]
+    email_address: str = Field(alias="emailAddress")
+    store_status: bool = Field(required=False, default=True)
+    latitude: float
+    longitude: float
+    location_code: str
+    timezone: str = Field(default=default_timezone)
+    account_id: int
+    account_name: str
+    client_id: int
+    client_name: str
+    country: str = Field(alias='country_id')
+    created_at: datetime = Field(default=datetime.now)
+    updated_at: datetime = Field(default=datetime.now)
+    store_type_id: StoreType
+    store_type: str = Field(alias="store_type_name")
+    visit_rule: List[str]
+    visit_category: List[str]
+    orgid: List[int] = Field(alias='orgids', default_factory=list)
+    custom_fields: List[CustomStoreField] = Field(default_factory=list)
+    client_id: List[int] = Field(alias='client_ids', default_factory=list)
+    client_name: List[str] = Field(alias='client_names', default_factory=list)
+    market_name: Optional[Dict[str, str]] = Field(required=False, alias='markets')
+    region_name: Optional[Dict[str, str]] = Field(required=False, alias='regions')
+    district_name: Optional[Dict[str, str]] = Field(required=False, alias='districts')
+    is_active: bool = Field(required=False, alias='store_is_active', default=True)
+
+    class Meta:
+        strict = True
+        as_objects = True
+        name = 'stores'
+        schema: str = 'networkninja'
+
+    def __post_init__(self):
+        super().__post_init__()
+        # geography
+        if self.market_name and isinstance(self.market_name, dict):
+            self.market_name = str(list(self.market_name.values())[0])
+        if self.region_name and isinstance(self.region_name, dict):
+            self.region_name = str(list(self.region_name.values())[0])
+        if self.district_name and isinstance(self.district_name, dict):
+            self.district_name = str(list(self.district_name.values())[0])
+        # client
+        if self.client_id and isinstance(self.client_id, list):
+            self.client_id = self.client_id[0]
+        if self.client_name and isinstance(self.client_name, list):
+            self.client_name = self.client_name[0]
+        # orgid:
+        if self.orgid and isinstance(self.orgid, list):
+            self.orgid = self.orgid[0]
+
+    async def _sync_object(self, conn):
+        """
+        Sync the organization with the Database.
+        """
+        query = f"""
+        select store_id, location_code FROM networkninja.stores where store_number = {self.store_number}
+        """
+        result = await conn.fetch_one(query)
+        if result:
+            self.store_id = result['store_id']
+            self.location_code = result['location_code']

flowtask/components/NetworkNinja/models/user.py

@@ -0,0 +1,151 @@
+
+import logging
+from typing import List, Optional
+from datetime import datetime
+from datamodel import BaseModel, Field
+from .abstract import AbstractPayload
+from .organization import Organization
+from .market import Market
+from .district import District
+from .region import Region
+from .client import Client
+
+
+class Role(AbstractPayload):
+    role_id: int = Field(primary_key=True, required=True)
+    client_id: int = Field(alias="role_client_id")
+    orgid: int = Field(alias="role_org_id")
+    role_name: str = Field(alias='name')
+    visit_id: int = Field(alias="role_visit_id")
+
+    class Meta:
+        strict = True
+        as_objects = True
+        name = 'roles'
+        schema: str = 'networkninja'
+
+class User(AbstractPayload):
+    """
+    User Model.
+
+    Represents a user in the system.
+
+    Example:
+        {
+            "user_id": 1,
+            "username": "admin",
+            "employee_number": 1234,
+            "first_name": "John",
+            "last_name": "Doe",
+            "email": "
+            "mobile_number": "123-456-7890",
+            "role_id": 1,
+            "role_name": "Admin",
+            "address": "1234 Elm St",
+            "city": "Springfield",
+            "state_code": "IL",
+            "zipcode": "62704",
+            "latitude": 39.781721,
+            "longitude": -89.650148,
+        }
+    """
+    user_id: int = Field(primary_key=True, required=True)
+    username: str = Field(required=True)
+    employee_number: int
+    first_name: str
+    last_name: str
+    display_name: str
+    email_address: str = Field(alias="email")
+    mobile_number: str
+    position_id: Optional[str]
+    role_id: int
+    role_name: str
+    address: str
+    city: str
+    state_code: str = Field(alias="state_name")
+    zipcode: str
+    latitude: Optional[float]
+    longitude: Optional[float]
+    physical_country: Optional[str]
+    is_active: bool = Field(required=True, default=True)
+    orgid: List[int]
+    client_id: List[int] = Field(alias='client_ids', default_factory=list)
+    client_name: List[str] = Field(alias='client_names', default_factory=list)
+    markets: List[Market] = Field(default_factory=list)
+    districts: List[District] = Field(default_factory=list)
+    regions: List[Region] = Field(default_factory=list)
+
+    class Meta:
+        strict = True
+        as_objects = True
+        name = 'users'
+        schema: str = 'networkninja'
+
+    def __post_init__(self):
+        super().__post_init__()
+        if not self.display_name:
+            self.display_name = f'{self.first_name} {self.last_name}'
+        if self.display_name and not self.first_name:
+            self.first_name, self.last_name = self.display_name.split(' ')
+
+
+class StaffingUser(AbstractPayload):
+    user_id: int = Field(primary_key=True, required=True, alias="id")
+    username: str = Field(required=True)
+    employee_number: str
+    first_name: str
+    last_name: str
+    display_name: str
+    email: str = Field(alias="email_address")
+    mobile_number: str
+    address: str
+    city: str
+    state_name: str = Field(alias="state")
+    physical_country: str
+    zipcode: str
+    position_id: Optional[str]
+    roles: List[Role] = Field(default_factory=list)
+    markets: List[Market] = Field(default_factory=list)
+    districts: List[District] = Field(default_factory=list)
+    regions: List[Region] = Field(default_factory=list)
+    orgid: List[int]
+    client_id: int
+    client_name: List[str] = Field(alias='client_names', default_factory=list)
+    clients: Optional[List[Client]] = Field(alias="client_ids", default_factory=list)
+    role_id: int
+    role_name: str
+    latitude: Optional[float]
+    longitude: Optional[float]
+    custom_fields: List[dict[str, str]]
+    onboarding: datetime
+    is_active: bool = Field(required=True, default=True)
+
+    class Meta:
+        strict = True
+        as_objects = True
+        name = 'staffing_users'
+        schema: str = 'networkninja'
+
+    def __post_init__(self):
+        super().__post_init__()
+        if not self.display_name:
+            self.display_name = f'{self.first_name} {self.last_name}'
+        if self.display_name and not self.first_name:
+            parts = self.display_name.split()
+            self.first_name, self.last_name = ' '.join(parts[:-1]), parts[-1]
+        if self.clients:
+            self.client_id = self.clients[0].client_id
+
+    # async def _sync_object(self, conn):
+    #     """
+    #     Sync Staffing User and dependencies.
+    #     """
+    #     # Sync Markets
+    #     markets = []
+    #     for market in self.markets:
+    #         market.market_id = int(market.market_id)
+    #         markets.append(market)
+    #     if markets:
+    #         await markets[0].update_many(
+    #             markets
+    #         )