repos
/ timestrap-django master

timestrap-django

mirror archived upstream

Time tracking you can host anywhere, built on Django. Full export support in multiple formats and easily extensible.

djangodockerhandcodedpythonself-hostedtime-trackingtimetracker

581 B · 22 lines · Python Raw History
 1from django.contrib.sites.managers import CurrentSiteManager as BaseCSM
 2
 3from .utils import current_site_id
 4
 5
 6class CurrentSiteManager(BaseCSM):
 7    """
 8    Replace the Django CurrentSiteManager to use current_site_id() instead of
 9    settings.SITE_ID.
10    """
11
12    def __init__(self, field_name=None, *args, **kwargs):
13        super().__init__(*args, **kwargs)
14        self.__field_name = field_name
15
16    def get_queryset(self):
17        return (
18            super()
19            .get_queryset()
20            .filter(**{self._get_field_name() + "__id": current_site_id()})
21        )