evyos-center-server/wag-managment-api-service-version-3
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

language_models updated

Browse Source
This commit is contained in:
berkay
2025-01-09 14:57:30 +03:00
parent 4fe1f36006
commit f4f9e584ff
26 changed files with 2346 additions and 187 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
databases/no_sql_models/identity.py
View File

@@ -1,6 +1,10 @@
import datetime
from databases.no_sql_models.validations import PasswordHistoryViaUser, DomainViaUser, AccessHistoryViaUser
from databases.no_sql_models.validations import (
PasswordHistoryViaUser,
DomainViaUser,
AccessHistoryViaUser,
)
from databases.no_sql_models.mongo_database import MongoQuery
@@ -39,7 +43,7 @@ class MongoQueryIdentity:
"other_domains_list": [payload.main_domain],
"main_domain": payload.main_domain,
"modified_at": datetime.datetime.now().timestamp(),
}
},
)
def update_domain_via_user(self, payload: DomainViaUser):
@@ -79,7 +83,7 @@ class MongoQueryIdentity:
raise Exception(
dict(
status_code=400,
detail="Password already used. Please enter a new password that you have not used last 3 times."
detail="Password already used. Please enter a new password that you have not used last 3 times.",
)
)

59
databases/no_sql_models/mongo_database.py
View File

@@ -28,9 +28,19 @@ class Paginate(BaseModel):
self.sortField,
-1 if self.sortOrder == "desc" else 1,
)
class MongoResponse:
def __init__(self, status: bool, message: str, message_code:str = None, data: dict = None, mongo=None, count: int = 1):
def __init__(
self,
status: bool,
message: str,
message_code: str = None,
data: dict = None,
mongo=None,
count: int = 1,
):
self.status = status
self.message = message
self.data = data
@@ -47,6 +57,7 @@ class MongoResponse:
"data": self.data,
}
class MongoQuery:
def __init__(self, table_name: str, database_name: str):
@@ -82,7 +93,7 @@ class MongoQuery:
message_code="NOT_CREATED",
data=None,
mongo=insert,
count=0
count=0,
)
def insert_many(self, payload) -> MongoResponse:
@@ -94,7 +105,7 @@ class MongoQuery:
message_code="CREATED",
data=payload,
mongo=insert_many,
count=len(payload)
count=len(payload),
)
return MongoResponse(
status=False,
@@ -102,7 +113,7 @@ class MongoQuery:
message_code="NOT_CREATED",
data=None,
mongo=insert_many,
count=0
count=0,
)
def find_or_insert(self, payload, field: str = "id") -> MongoResponse:
@@ -120,7 +131,7 @@ class MongoQuery:
status=True,
message="The document has been created",
data=payload,
mongo=mongo
mongo=mongo,
)
found_row = self.get_one(match=payload[field], field=field)
@@ -130,7 +141,7 @@ class MongoQuery:
message_code="ALREADY_EXISTS",
message="The document already exists",
data=found_row.data,
mongo=found_row.mongo
mongo=found_row.mongo,
)
insert_row = self.table.insert_one(document=payload)
return MongoResponse(
@@ -138,7 +149,7 @@ class MongoQuery:
message="The document has been created",
message_code="CREATED",
data=payload,
mongo=insert_row
mongo=insert_row,
)
def update(self, match, payload, field: str = "id"):
@@ -150,15 +161,17 @@ class MongoQuery:
message="The document has been updated",
message_code="UPDATED",
data=self.get_one(match=match, field=field).data,
mongo=update_one
mongo=update_one,
)
update_one = self.table.update_one(filter={field: match}, update={"$set": payload})
update_one = self.table.update_one(
filter={field: match}, update={"$set": payload}
)
return MongoResponse(
status=True,
message="The document has been updated",
message_code="UPDATED",
data=self.get_one(match=match, field=field).data,
mongo=update_one
mongo=update_one,
)
def get_one(self, match, field: str = "id"):
@@ -169,7 +182,7 @@ class MongoQuery:
message="The document has been found",
message_code="FOUND",
data=self.parse_json(data=get_one),
mongo=get_one
mongo=get_one,
)
if get_one := self.table.find_one(filter={field: match}):
return MongoResponse(
@@ -177,14 +190,14 @@ class MongoQuery:
message="The document has been found",
message_code="FOUND",
data=self.parse_json(data=get_one),
mongo=get_one
mongo=get_one,
)
return MongoResponse(
status=False,
message="The document has not been found",
message_code="NOT_FOUND",
data=None,
mongo=None
mongo=None,
)
def filter_by(self, payload, sort_by: str = "_id", sort_direction: str = "asc"):
@@ -202,7 +215,7 @@ class MongoQuery:
message_code="NOT_FOUND",
data=None,
mongo=return_,
count=0
count=0,
)
return MongoResponse(
status=True,
@@ -222,7 +235,7 @@ class MongoQuery:
message_code="DELETED",
data=None,
count=delete_one.deleted_count,
mongo=delete_one
mongo=delete_one,
)
delete_one = self.table.delete_one(filter={field: match})
if delete_one.deleted_count:
@@ -232,7 +245,7 @@ class MongoQuery:
message_code="DELETED",
data=None,
count=delete_one.deleted_count,
mongo=delete_one
mongo=delete_one,
)
return MongoResponse(
status=False,
@@ -240,7 +253,7 @@ class MongoQuery:
message_code="NOT_DELETED",
data=None,
mongo=None,
count=0
count=0,
)
def delete_all(self, match, field: str = "id"):
@@ -253,7 +266,7 @@ class MongoQuery:
message_code="DELETED",
data=None,
count=delete_many.deleted_count,
mongo=delete_many
mongo=delete_many,
)
delete_many = self.table.delete_many(filter={field: match})
if delete_many.deleted_count:
@@ -263,7 +276,7 @@ class MongoQuery:
message_code="DELETED",
data=None,
count=delete_many.deleted_count,
mongo=delete_many
mongo=delete_many,
)
return MongoResponse(
status=False,
@@ -271,7 +284,7 @@ class MongoQuery:
message_code="NOT_DELETED",
data=None,
mongo=None,
count=0
count=0,
)
def list_all(self, paginate: Paginate):
@@ -286,7 +299,7 @@ class MongoQuery:
message_code="NOT_FOUND",
data=None,
mongo=return_,
count=0
count=0,
)
return MongoResponse(
status=True,
@@ -294,7 +307,7 @@ class MongoQuery:
message_code="FOUND",
count=self.table.count_documents({}),
data=data,
mongo=return_
mongo=return_,
)
def get_all(self):
@@ -305,7 +318,7 @@ class MongoQuery:
message_code="FOUND",
count=self.table.count_documents({}),
data=self.parse_json(data=return_),
mongo=return_
mongo=return_,
)