diff options
author | Frederick Muriuki Muriithi | 2023-09-13 08:01:50 +0300 |
---|---|---|
committer | Frederick Muriuki Muriithi | 2023-09-26 03:44:28 +0300 |
commit | 9a3b628e2744b658ade8902c1cd33deccbd48b63 (patch) | |
tree | 47299176e0c2052f24060a9596417b10a16e2c7c /gn_auth/auth | |
parent | a49f925eaaff5b29ffc387a7f02023e111208d5f (diff) | |
download | gn-auth-9a3b628e2744b658ade8902c1cd33deccbd48b63.tar.gz |
Extract basic resource types to a separate module.
Diffstat (limited to 'gn_auth/auth')
-rw-r--r-- | gn_auth/auth/authorisation/resources/base.py | 37 | ||||
-rw-r--r-- | gn_auth/auth/authorisation/resources/models.py | 34 |
2 files changed, 38 insertions, 33 deletions
diff --git a/gn_auth/auth/authorisation/resources/base.py b/gn_auth/auth/authorisation/resources/base.py new file mode 100644 index 0000000..d2075c2 --- /dev/null +++ b/gn_auth/auth/authorisation/resources/base.py @@ -0,0 +1,37 @@ +"""Base types for resources.""" +from uuid import UUID +from typing import Any, Sequence, NamedTuple + +from gn_auth.auth.dictify import dictify + +class ResourceCategory(NamedTuple): + """Class representing a resource category.""" + resource_category_id: UUID + resource_category_key: str + resource_category_description: str + + def dictify(self) -> dict[str, Any]: + """Return a dict representation of `ResourceCategory` objects.""" + return { + "resource_category_id": self.resource_category_id, + "resource_category_key": self.resource_category_key, + "resource_category_description": self.resource_category_description + } + +class Resource(NamedTuple): + """Class representing a resource.""" + resource_id: UUID + resource_name: str + resource_category: ResourceCategory + public: bool + resource_data: Sequence[dict[str, Any]] = tuple() + + def dictify(self) -> dict[str, Any]: + """Return a dict representation of `Resource` objects.""" + return { + "resource_id": self.resource_id, + "resource_name": self.resource_name, + "resource_category": dictify(self.resource_category), + "public": self.public, + "resource_data": self.resource_data + } diff --git a/gn_auth/auth/authorisation/resources/models.py b/gn_auth/auth/authorisation/resources/models.py index e4c3cab..5718753 100644 --- a/gn_auth/auth/authorisation/resources/models.py +++ b/gn_auth/auth/authorisation/resources/models.py @@ -3,7 +3,7 @@ import json import sqlite3 from uuid import UUID, uuid4 from functools import reduce, partial -from typing import Any, Dict, Sequence, Optional, NamedTuple +from typing import Dict, Sequence, Optional from ...db import sqlite3 as db from ...dictify import dictify @@ -20,38 +20,6 @@ from ..groups.models import ( class MissingGroupError(AuthorisationError): """Raised for any resource operation without a group.""" -class ResourceCategory(NamedTuple): - """Class representing a resource category.""" - resource_category_id: UUID - resource_category_key: str - resource_category_description: str - - def dictify(self) -> dict[str, Any]: - """Return a dict representation of `ResourceCategory` objects.""" - return { - "resource_category_id": self.resource_category_id, - "resource_category_key": self.resource_category_key, - "resource_category_description": self.resource_category_description - } - -class Resource(NamedTuple): - """Class representing a resource.""" - group: Group - resource_id: UUID - resource_name: str - resource_category: ResourceCategory - public: bool - resource_data: Sequence[dict[str, Any]] = tuple() - - def dictify(self) -> dict[str, Any]: - """Return a dict representation of `Resource` objects.""" - return { - "group": dictify(self.group), "resource_id": self.resource_id, - "resource_name": self.resource_name, - "resource_category": dictify(self.resource_category), - "public": self.public, - "resource_data": self.resource_data - } def __assign_resource_owner_role__(cursor, resource, user): """Assign `user` the 'Resource Owner' role for `resource`.""" |