accrete 0.0.39__tar.gz → 0.0.40__tar.gz

{accrete-0.0.39 → accrete-0.0.40}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: accrete
-Version: 0.0.39
+Version: 0.0.40
 Summary: Django Shared Schema Multi Tenant
 Author-email: Benedikt Jilek <benedikt.jilek@pm.me>
 License: Copyright (c) 2023 Benedikt Jilek

{accrete-0.0.39 → accrete-0.0.40}/accrete/annotation.py

@@ -1,4 +1,4 @@
-from django.db.models import Field, QuerySet
+from django.db.models import Field
 from django.db.models.expressions import Func
 from django.db.models.aggregates import Aggregate
 
@@ -34,12 +34,10 @@ class AnnotationModelMixin:
 
 class AnnotationManagerMixin:
 
-    def add_annotations(self, queryset: QuerySet):
-        model = self.model
-        if not hasattr(model, 'get_annotations'):
-            return queryset
-        return queryset.annotate(**{
+    def get_annotations(self):
+        if not hasattr(self.model, 'get_annotations'):
+            return {}
+        return {
             annotation['name']: annotation['annotation'].function
-            for annotation in model.get_annotations()
-        })
-
+            for annotation in self.model.get_annotations()
+        }

{accrete-0.0.39 → accrete-0.0.40}/accrete/contrib/user/views.py

@@ -1,15 +1,12 @@
-from django.utils import timezone
 from django.contrib.auth.forms import AuthenticationForm
 from django.contrib.auth import views, update_session_auth_hash
 from django.contrib.auth.decorators import login_required
-from django.contrib.sessions.models import Session
 from django.contrib import messages
-from django.db.models import Q
 from django.shortcuts import redirect, render, reverse, resolve_url
 from django.utils.translation import gettext_lazy as _
 from django.conf import settings
 
-from accrete.forms import save_form
+from accrete.utils import save_form
 from accrete.contrib import ui
 from .forms import UserForm, ChangePasswordForm, ChangeEmailForm
 

{accrete-0.0.39 → accrete-0.0.40}/accrete/forms.py

@@ -290,22 +290,3 @@ class TenantModelForm(One2ManyModelForm):
             self.save_o2m()
 
         return self.instance
-
-
-def save_form(form, reraise=False):
-    form.is_saved = False
-    form.save_error = None
-    form.save_error_id = None
-    try:
-        if form.is_valid():
-            with transaction.atomic():
-                form.save()
-            form.is_saved = True
-    except Exception as e:
-        form.save_error = repr(e)
-        error_id = str(uuid4())[:8]
-        _logger.exception(f'{error_id}: {e}')
-        form.save_error_id = error_id
-        if reraise:
-            raise e
-    return form

accrete-0.0.40/accrete/managers.py

@@ -0,0 +1,35 @@
+from django.db import models
+from accrete.tenant import get_tenant
+from accrete.annotation import AnnotationManagerMixin
+
+
+class TenantManager(models.Manager, AnnotationManagerMixin):
+
+    def get_queryset(self):
+        queryset = super().get_queryset()
+        tenant = get_tenant()
+        if tenant:
+            queryset = queryset.filter(tenant=tenant)
+        return queryset.annotate(**self.get_annotations())
+
+    def bulk_create(
+            self,
+            objs,
+            batch_size=None,
+            ignore_conflicts=False,
+            update_conflicts=False,
+            update_fields=None,
+            unique_fields=None,
+    ):
+        tenant = get_tenant()
+        if tenant is None and not all(obj.tenant_id for obj in objs):
+            raise ValueError(
+                'Tenant must be set for all objects when calling bulk_create'
+            )
+        for obj in objs:
+            obj.tenant_id = tenant.pk
+        return super().bulk_create(
+            objs, batch_size=batch_size, ignore_conflicts=ignore_conflicts,
+            update_conflicts=update_conflicts, update_fields=update_fields,
+            unique_fields=unique_fields
+        )

{accrete-0.0.39 → accrete-0.0.40}/accrete/models.py

@@ -3,39 +3,8 @@ from django.conf import settings
 from django.utils.translation import gettext_lazy as _
 from django.contrib.auth.validators import UnicodeUsernameValidator
 from accrete.tenant import get_tenant
-from accrete.annotation import AnnotationModelMixin, AnnotationManagerMixin
-
-
-class TenantManager(models.Manager, AnnotationManagerMixin):
-
-    def get_queryset(self):
-        queryset = super().get_queryset()
-        tenant = get_tenant()
-        if tenant:
-            queryset = queryset.filter(tenant=tenant)
-        return self.add_annotations(queryset)
-
-    def bulk_create(
-            self,
-            objs,
-            batch_size=None,
-            ignore_conflicts=False,
-            update_conflicts=False,
-            update_fields=None,
-            unique_fields=None,
-    ):
-        tenant = get_tenant()
-        if tenant is None and not all(obj.tenant_id for obj in objs):
-            raise ValueError(
-                'Tenant must be set for all objects when calling bulk_create'
-            )
-        for obj in objs:
-            obj.tenant_id = tenant.pk
-        return super().bulk_create(
-            objs, batch_size=batch_size, ignore_conflicts=ignore_conflicts,
-            update_conflicts=update_conflicts, update_fields=update_fields,
-            unique_fields=unique_fields
-        )
+from accrete.annotation import AnnotationModelMixin
+from accrete.managers import TenantManager
 
 
 class TenantModel(models.Model, AnnotationModelMixin):

accrete-0.0.40/accrete/utils/__init__.py

@@ -0,0 +1,4 @@
+from . import dates
+from .forms import save_form
+from .http import filter_from_querystring
+from .models import get_related_model

accrete-0.0.40/accrete/utils/forms.py

@@ -0,0 +1,24 @@
+import logging
+from uuid import uuid4
+from django.db import transaction
+
+_logger = logging.getLogger(__name__)
+
+
+def save_form(form, reraise=False):
+    form.is_saved = False
+    form.save_error = None
+    form.save_error_id = None
+    try:
+        if form.is_valid():
+            with transaction.atomic():
+                form.save()
+            form.is_saved = True
+    except Exception as e:
+        form.save_error = repr(e)
+        error_id = str(uuid4())[:8]
+        _logger.exception(f'{error_id}: {e}')
+        form.save_error_id = error_id
+        if reraise:
+            raise e
+    return form

accrete-0.0.39/accrete/querystring.py → accrete-0.0.40/accrete/utils/http.py

@@ -7,14 +7,24 @@ from accrete.annotation import Annotation
 
 _logger = logging.getLogger(__name__)
 
+Querystring_KEY_MAP = {
+    'querystring': 'q',
+    'order': 'order',
+    'paginate_by': 'paginate_by',
+    'page': 'page'
+}
+
 
 def filter_from_querystring(
-        model: type[Model], query_string: str, order: list[str] = None
+        model: type[Model], get_params: dict, key_map: dict = None
 ) -> QuerySet:
 
-    order = order or model._meta.ordering
+    key_map = key_map or Querystring_KEY_MAP
+    querystring = get_params.get(key_map['querystring'], '[]')
+    order = get_params.get(key_map['order']) or model._meta.ordering
+
     return model.objects.filter(
-        parse_querystring(model, query_string)
+        parse_querystring(model, querystring)
     ).order_by(*order).distinct()
 
 
@@ -86,4 +96,4 @@ def parse_querystring(model: type[Model], query_string: str) -> Q:
 
     ops = {'&': operator.and_, '|': operator.or_, '^': operator.xor}
     query = parse_query_block(query_data)
-    return query
+    return query

{accrete-0.0.39 → accrete-0.0.40}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = 'hatchling.build'
 
 [project]
 name = 'accrete'
-version = '0.0.39'
+version = '0.0.40'
 authors = [
   { name='Benedikt Jilek', email='benedikt.jilek@pm.me' },
 ]

accrete-0.0.39/accrete/queries.py

@@ -1,18 +0,0 @@
-import logging
-from accrete import models
-from accrete.tenant import get_tenant
-
-_logger = logging.getLogger(__name__)
-
-
-def is_member(tenant, user):
-    return tenant.members.filter(user=user, is_active=True).exists()
-
-
-def members_for_current_tenant():
-    tenant = get_tenant()
-    return tenant and tenant.members or models.Member.objects.none()
-
-
-def all_tenants():
-    return models.Tenant.objects.all()