def limit_choices(self,
site=None,
controller=None,
feed=None,
envgroup=None,
**_):
gu = get_profile(get_request())
if feed is not None:
self.fields['feed'].queryset = gu.feeds.filter(id=feed.id)
elif site is not None:
self.fields['feed'].queryset = gu.feeds.filter(site=site)
if controller is not None:
self.fields['controller'].queryset = gu.controllers.filter(
id=controller.id)
elif site is not None:
self.fields['controller'].queryset = gu.controllers.filter(
site=site)
if envgroup is not None:
self.fields['envgroup'].queryset = gu.envgroups.filter(
id=envgroup.id)
elif site is not None:
self.fields['envgroup'].queryset = gu.envgroups.filter(site=site)
if site is not None:
self.fields['groups'].queryset = gu.groups.filter(site=site)
else:
self.fields['groups'].queryset = gu.groups ## should not happen
def limit_choices(self, site=None, valve=None, **_):
gu = get_profile(get_request())
v = gu.all_valves
if valve is not None:
v = v.filter(id=valve.id)
if site is not None:
v = v.filter(controller__site__id=site.id)
self.fields['valves'].queryset = v
def limit_choices(self,site=None,valve=None): gu = get_profile(get_request()) v = gu.all_valves if valve is not None: v = v.filter(id=valve.id) if site is not None: v = v.filter(controller__site__id=site.id) self.fields['valves'].queryset = v
def get_context_data(self,**k):
ctx = super(ValveOverrideView,self).get_context_data(**k)
gu = get_profile(get_request())
av = limit_choices(gu.all_valves,**self.aux_data)
q = ValveOverride.objects.filter(valve__in=av)
try:
ctx['next_s'] = q.filter(start__gt=ctx['valveoverride'].start).order_by('start')[0]
except IndexError:
ctx['next_s'] = None
try:
ctx['prev_s'] = q.filter(start__lt=ctx['valveoverride'].start).order_by('-start')[0]
except IndexError:
ctx['prev_s'] = None
return ctx
def get_context_data(self, **k):
ctx = super(ScheduleView, self).get_context_data(**k)
gu = get_profile(get_request())
av = limit_choices(gu.all_valves, **self.aux_data)
q = Schedule.objects.filter(valve__in=av)
try:
ctx['next_s'] = q.filter(
start__gt=ctx['schedule'].start).order_by('start')[0]
except IndexError:
ctx['next_s'] = None
try:
ctx['prev_s'] = q.filter(
start__lt=ctx['schedule'].start).order_by('-start')[0]
except IndexError:
ctx['prev_s'] = None
return ctx
def get_context_data(self, **k):
ctx = super(GroupOverrideView, self).get_context_data(**k)
gu = get_profile(get_request())
av = limit_choices(gu.groups, **self.aux_data)
q = GroupOverride.objects.filter(group__in=av)
try:
ctx['next_s'] = q.filter(
start__gt=ctx['groupoverride'].start).order_by('start')[0]
except IndexError:
ctx['next_s'] = None
try:
ctx['prev_s'] = q.filter(
start__lt=ctx['groupoverride'].start).order_by('-start')[0]
except IndexError:
ctx['prev_s'] = None
return ctx
def limit_choices(self,site=None,controller=None,feed=None,envgroup=None, **_): gu = get_profile(get_request()) if feed is not None: self.fields['feed'].queryset = gu.feeds.filter(id=feed.id) elif site is not None: self.fields['feed'].queryset = gu.feeds.filter(site=site) if controller is not None: self.fields['controller'].queryset = gu.controllers.filter(id=controller.id) elif site is not None: self.fields['controller'].queryset = gu.controllers.filter(site=site) if envgroup is not None: self.fields['envgroup'].queryset = gu.envgroups.filter(id=envgroup.id) elif site is not None: self.fields['envgroup'].queryset = gu.envgroups.filter(site=site) if site is not None: self.fields['groups'].queryset = gu.groups.filter(site=site) else: self.fields['groups'].queryset = gu.groups ## should not happen
def get_queryset(self):
gu = get_profile(self.request)
return super(ControllerMixin,
self).get_queryset().filter(id__in=gu.controllers)
def get_queryset(self): gu = get_profile(self.request) return super(FeedMixin,self).get_queryset().filter(id__in=gu.feeds)
def get_queryset(self):
gu = get_profile(self.request)
return super(ValveMixin,
self).get_queryset().filter(id__in=gu.all_valves)
def get_queryset(self): gu = get_profile(self.request) return super(SiteMixin,self).get_queryset().filter(id__in=gu.sites.all())
def get_queryset(self):
gu = get_profile(self.request)
return super(GroupOverrideMixin,self).get_queryset().filter(group__site__id__in=gu.sites.all()).order_by("-start")
def get_queryset(self):
gu = get_profile(self.request)
return super(
EnvItemMixin,
self).get_queryset().filter(group__site__id__in=gu.sites.all())
def get_queryset(self):
gu = get_profile(self.request)
return super(EnvGroupMixin,
self).get_queryset().filter(id__in=gu.envgroups)
def get_queryset(self):
gu = get_profile(self.request)
return super(FeedMixin, self).get_queryset().filter(id__in=gu.feeds)
def get_queryset(self):
gu = get_profile(self.request)
return super(HistoryMixin,self).get_queryset().filter(site__id__in=gu.sites.all()).order_by("-time")
def get_queryset(self):
gu = get_profile(self.request)
return super(LevelMixin, self).get_queryset().filter(
valve__in=gu.all_valves).order_by('-time')
def get_queryset(self): gu = get_profile(self.request) return super(EnvItemMixin,self).get_queryset().filter(group__site__id__in=gu.sites.all())
def get_queryset(self): gu = get_profile(self.request) return super(ControllerMixin,self).get_queryset().filter(id__in=gu.controllers)
def get_queryset(self): gu = get_profile(self.request) return super(ValveMixin,self).get_queryset().filter(id__in=gu.all_valves)
def limit_choices(self,**k): gu = get_profile(get_request()) self.fields['valve'].queryset = limit_choices(gu.all_valves,**k)
def limit_choices(self,**k): gu = get_profile(get_request()) self.fields['group'].queryset = limit_choices(gu.groups,**k)
def get_queryset(self):
gu = get_profile(self.request)
return super(GroupOverrideMixin, self).get_queryset().filter(
group__site__id__in=gu.sites.all()).order_by("-start")
def get_queryset(self): gu = get_profile(self.request) return super(EnvGroupMixin,self).get_queryset().filter(id__in=gu.envgroups)
def save(self,commit=True): r = super(SiteForm,self).save(commit) gu = get_profile(get_request()) gu.sites.add(r) gu.save() ## TODO: check if 'commit' is True! return r
def get_queryset(self):
gu = get_profile(self.request)
return super(ValveOverrideMixin,self).get_queryset().filter(valve__controller__site__id__in=gu.sites.all()).order_by("-start")
def get_queryset(self):
gu = get_profile(self.request)
return super(LevelMixin,self).get_queryset().filter(valve__in=gu.all_valves).order_by('-time')
def limit_choices(self, **k):
gu = get_profile(get_request())
self.fields['group'].queryset = limit_choices(gu.groups, **k)
def get_queryset(self):
gu = get_profile(self.request)
return super(LogMixin, self).get_queryset().filter(
site__id__in=gu.sites.all()).order_by("-timestamp")
def get_queryset(self):
gu = get_profile(self.request)
return super(ScheduleMixin, self).get_queryset().filter(
valve__controller__site__id__in=gu.sites.all()).order_by("-start")
def save(self,commit=True): r = super(SiteForm,self).save(commit) gu = get_profile(get_request()) gu.sites.add(r) gu.save() ## TODO: check if 'commit' is True! return r
def limit_choices(self, **k):
gu = get_profile(get_request())
self.fields['valve'].queryset = limit_choices(gu.all_valves, **k)
def get_queryset(self): gu = get_profile(self.request) return super(SiteMixin,self).get_queryset().filter(id__in=gu.sites.all())
