201 lines
7.1 KiB
Python
201 lines
7.1 KiB
Python
"""
|
|
|
|
AUTHOR:
|
|
|
|
Khushal P Soonderji
|
|
|
|
DATE:
|
|
|
|
Saturday, 7th Dec., 2024.
|
|
|
|
OBJECTIVE:
|
|
|
|
To define how auth tokens will be stored in the database.
|
|
|
|
REFERENCES:
|
|
|
|
N/A
|
|
|
|
DOWNLOADS:
|
|
|
|
N/A
|
|
|
|
"""
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** IMPORT ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
# To make sibling directories accessible for imports:
|
|
import sys
|
|
sys.path.append(".")
|
|
sys.path.append("..")
|
|
|
|
# For making data behaviour_models:
|
|
from pydantic import BaseModel, Field, field_validator, PastDatetime, AwareDatetime
|
|
from typing import Optional, Literal, Union
|
|
|
|
# My utils:
|
|
from utils_v2.string import regex
|
|
from utils_v2.date_time import date_time
|
|
|
|
# To work with MongoDB:
|
|
from bson.objectid import ObjectId
|
|
|
|
# To work with date and time:
|
|
import datetime
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** MACROS / ONE-TIME INIT ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
# --- Nothing Yet
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** VARIABLES ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
# --- Nothing Yet
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** FUNCTIONS ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
class CoreAuthTokenModel(BaseModel):
|
|
|
|
version: str = Field(
|
|
description = "a hint about the version no. of this message",
|
|
min_length = 1,
|
|
frozen = True,
|
|
default = "1.0.0"
|
|
)
|
|
|
|
serviceType: Literal["email", "sms", "chat"] = Field(
|
|
description = "the kind of service this message was sent/received from",
|
|
frozen = True
|
|
)
|
|
|
|
client: Literal[
|
|
"gmail", "outlook", # ...................... Mail Clients
|
|
"telegram", "whatsapp", # .................. Chat Clients
|
|
"nimbusSmsIndia", "savvyBulkSmsKenya", # ... SMS Clients
|
|
"razorpay", "safaricomMPesaExpress" # ...... Payment Gateways
|
|
] = Field(
|
|
description = "the third-part client that was used",
|
|
frozen = True
|
|
)
|
|
|
|
authType: Literal["oauth", "auth"] = Field(
|
|
description = "the type of authentication procedure used",
|
|
frozen = True
|
|
)
|
|
|
|
firstRequestTs: AwareDatetime = Field(
|
|
description = "the time (utc) at which authorization was first requested",
|
|
frozen = True
|
|
)
|
|
|
|
lastRequestTs: AwareDatetime = Field(
|
|
description = "the time (utc) at which authorization was last requested",
|
|
frozen = False
|
|
)
|
|
|
|
firstRefreshTs: AwareDatetime = Field(
|
|
description = "the time (utc) at which the tokens were first refreshed",
|
|
frozen = False
|
|
)
|
|
|
|
lastRefreshTs: AwareDatetime = Field(
|
|
description = "the time (utc) at which the tokens were last refreshed",
|
|
frozen = False
|
|
)
|
|
|
|
token: dict | None = Field(
|
|
description = "the actual auth tokens of that client; will differ for each client",
|
|
frozen = True,
|
|
default_factory = lambda: date_time.get_current_utc_date_time(as_string = False)
|
|
)
|
|
|
|
user: dict = Field(
|
|
description = "how you identify your user",
|
|
frozen = True
|
|
)
|
|
|
|
clientUserId: dict = Field(
|
|
description = "how third-party client identifies the same user",
|
|
frozen = True
|
|
)
|
|
|
|
# ┏┓ ┏•
|
|
# ┃ ┏┓┏┓╋┓┏┓
|
|
# ┗┛┗┛┛┗┛┗┗┫
|
|
# ┛
|
|
|
|
class Config:
|
|
extra = "allow"
|
|
arbitrary_types_allowed = True
|
|
|
|
# ┓┏ ┓• ┓ •
|
|
# ┃┃┏┓┃┓┏┫┏┓╋┓┏┓┏┓
|
|
# ┗┛┗┻┗┗┗┻┗┻┗┗┗┛┛┗
|
|
|
|
@field_validator(
|
|
"firstRequestTs",
|
|
"lastRequestTs", "firstRefreshTs", "lastRefreshTs",
|
|
mode = "before"
|
|
)
|
|
def parse_date_time(cls, value):
|
|
return date_time.parse_date_time(input_value = value, timezone = date_time.TIMEZONE_UTC)
|
|
|
|
|
|
# *****************************************************************************************************************
|
|
# ***** ****
|
|
# *** MAIN PROGRAM ***
|
|
# ***** ****
|
|
# *****************************************************************************************************************
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from utils_v2.string import json
|
|
|
|
auth_token = CoreAuthTokenModel(
|
|
serviceType = "email",
|
|
client = "gmail",
|
|
authType = "oauth",
|
|
firstRequestTs = date_time.get_current_utc_date_time(as_string = False),
|
|
lastRequestTs = date_time.get_current_utc_date_time(as_string = False),
|
|
firstRefreshTs = date_time.get_current_utc_date_time(as_string = False),
|
|
lastRefreshTs = date_time.get_current_utc_date_time(as_string = False),
|
|
token = {
|
|
"username": "testing123",
|
|
"password": "abcdefgh"
|
|
},
|
|
user = {
|
|
"userId": 0,
|
|
"entityId": 1,
|
|
"billingAccountId": 2,
|
|
"fullName": "Bhopli"
|
|
},
|
|
clientUserId = {
|
|
"email": "bhopli@gmail.com"
|
|
}
|
|
)
|
|
|
|
print("AUTH-TOKEN MODEL:", json.to_string(auth_token.model_dump(), default = str))
|