simo 2.0.42__py3-none-any.whl → 2.1.2__py3-none-any.whl

simo/asgi.py

@@ -21,7 +21,7 @@ for name, app in apps.app_configs.items():
         if isinstance(item, list) and var_name == 'urlpatterns':
             urlpatterns.extend(item)
 
-from .on_http_start import *
+from .management.on_http_start import *
 
 application = ProtocolTypeRouter({
     "http": get_asgi_application(),

simo/core/__init__.py

@@ -0,0 +1 @@
+default_app_config = 'core.apps.SIMOCoreAppConfig'

simo/core/admin.py

@@ -1,6 +1,8 @@
+import markdown
 from django.utils.translation import gettext_lazy as _
 from django.contrib import admin
 from django.urls import reverse
+from django.utils.safestring import mark_safe
 from easy_thumbnails.fields import ThumbnailerField
 from adminsortable2.admin import SortableAdminMixin
 from django.template.loader import render_to_string
@@ -8,6 +10,7 @@ from django.utils.decorators import method_decorator
 from django.views.decorators.csrf import csrf_protect
 from django.shortcuts import redirect, render
 from simo.users.models import ComponentPermission
+from simo.core.utils.admin import EasyObjectsDeleteMixin
 from .utils.type_constants import (
     ALL_BASE_TYPES, GATEWAYS_MAP, CONTROLLERS_BY_GATEWAY
 )
@@ -27,7 +30,7 @@ csrf_protect_m = method_decorator(csrf_protect)
 
 
 @admin.register(Icon)
-class IconAdmin(admin.ModelAdmin):
+class IconAdmin(EasyObjectsDeleteMixin, admin.ModelAdmin):
     form = IconForm
     list_display = 'slug', 'preview', 'copyright'
     search_fields = 'slug', 'keywords',
@@ -57,25 +60,8 @@ class IconAdmin(admin.ModelAdmin):
 
 
 
-@admin.register(Instance)
-class InstanceAdmin(admin.ModelAdmin):
-    list_display = 'name', 'timezone', 'uid'
-    exclude = 'learn_fingerprints_start', 'learn_fingerprints'
-
-
-    def has_module_permission(self, request):
-        return request.user.is_master
-
-    def has_view_permission(self, request, obj=None):
-        return self.has_module_permission(request)
-
-    def has_change_permission(self, request, obj=None):
-        return self.has_module_permission(request)
-
-
-
 @admin.register(Zone)
-class ZoneAdmin(SortableAdminMixin, admin.ModelAdmin):
+class ZoneAdmin(EasyObjectsDeleteMixin, SortableAdminMixin, admin.ModelAdmin):
     list_display = 'name', 'instance'
     search_fields = 'name',
     list_filter = 'instance',
@@ -91,7 +77,7 @@ class ZoneAdmin(SortableAdminMixin, admin.ModelAdmin):
 
 
 @admin.register(Category)
-class CategoryAdmin(SortableAdminMixin, admin.ModelAdmin):
+class CategoryAdmin(EasyObjectsDeleteMixin, SortableAdminMixin, admin.ModelAdmin):
     form = CategoryAdminForm
     list_display = 'name_display', 'all'
     search_fields = 'name',
@@ -118,7 +104,7 @@ class CategoryAdmin(SortableAdminMixin, admin.ModelAdmin):
 
 
 @admin.register(Gateway)
-class GatewayAdmin(admin.ModelAdmin):
+class GatewayAdmin(EasyObjectsDeleteMixin, admin.ModelAdmin):
     list_display = 'type', 'status'
     readonly_fields = ('type', 'control')
 
@@ -254,14 +240,14 @@ class ComponentPermissionInline(admin.TabularInline):
 
 
 @admin.register(Component)
-class ComponentAdmin(admin.ModelAdmin):
+class ComponentAdmin(EasyObjectsDeleteMixin, admin.ModelAdmin):
     form = BaseComponentForm
     list_display = (
         'id', 'name_display', 'value_display', 'base_type', 'alive', 'battery_level',
         'alarm_category', 'show_in_app',
     )
     readonly_fields = (
-        'id', 'controller_uid', 'base_type', 'gateway', 'config',
+        'id', 'controller_uid', 'base_type', 'info', 'gateway', 'config',
         'alive', 'error_msg', 'battery_level',
         'control', 'value', 'arm_status', 'history', 'meta'
     )
@@ -274,6 +260,9 @@ class ComponentAdmin(admin.ModelAdmin):
     list_per_page = 100
     change_list_template = 'admin/component_change_list.html'
     inlines = ComponentPermissionInline,
+    # standard django admin change_form.html template + adds side panel
+    # for displaying component controller info.
+    #change_form_template = 'admin/core/component_change_form.html'
 
     def get_fieldsets(self, request, obj=None):
         form = self._get_form_for_get_fields(request, obj)
@@ -346,6 +335,7 @@ class ComponentAdmin(admin.ModelAdmin):
                 ctx['selected_type'] = ALL_BASE_TYPES.get(
                     controller_cls.base_type, controller_cls.base_type
                 )
+                ctx['info'] = controller_cls.info(controller_cls)
                 if request.method == 'POST':
                     ctx['form'] = add_form(
                         request=request,
@@ -356,7 +346,6 @@ class ComponentAdmin(admin.ModelAdmin):
                     pop_fields_from_form(ctx['form'])
                     if ctx['form'].is_valid():
                         if ctx['form'].controller.is_discoverable:
-                            print("INIT DISCOVERY!!!")
                             ctx['form'].controller.init_discovery(
                                 ctx['form'].cleaned_data
                             )
@@ -395,7 +384,6 @@ class ComponentAdmin(admin.ModelAdmin):
                     if ctx['form'].is_valid():
                         request.session['add_comp_type'] = \
                             ctx['form'].cleaned_data['controller_type']
-                        print("Session controller type: ", request.session['add_comp_type'])
                         return redirect(request.path)
 
                 else:
@@ -475,6 +463,18 @@ class ComponentAdmin(admin.ModelAdmin):
             }
         )
 
+    def info(self, obj):
+        if not obj.controller:
+            return
+        info = obj.controller.info()
+        if not info:
+            return
+        return mark_safe(
+            f'<div class="markdownified-info">'
+            f'{markdown.markdown(info)}'
+            f'</div>'
+        )
+
     def history(self, obj):
         if not obj:
             return ''
@@ -483,4 +483,4 @@ class ComponentAdmin(admin.ModelAdmin):
                 'value_history': obj.history.filter(type='value').order_by('-date')[:50],
                 'arm_status_history': obj.history.filter(type='security').order_by('-date')[:50]
             }
-        )
+        )

simo/core/api.py

@@ -9,6 +9,7 @@ from django.utils import timezone
 from django.http import HttpResponse, Http404
 from simo.core.utils.helpers import get_self_ip, search_queryset
 from rest_framework import status
+from actstream.models import Action
 from rest_framework.pagination import PageNumberPagination
 from rest_framework import viewsets
 from django_filters.rest_framework import DjangoFilterBackend
@@ -23,7 +24,8 @@ from .models import (
 )
 from .serializers import (
     IconSerializer, CategorySerializer, ZoneSerializer,
-    ComponentSerializer, ComponentHistorySerializer
+    ComponentSerializer, ComponentHistorySerializer,
+    ActionSerializer
 )
 from .permissions import (
     IsInstanceSuperuser, InstanceSuperuserCanEdit, ComponentPermission
@@ -494,6 +496,22 @@ class ComponentHistoryViewSet(InstanceMixin, viewsets.ReadOnlyModelViewSet):
         return vectors
 
 
+class ActionsViewset(InstanceMixin, viewsets.ReadOnlyModelViewSet):
+    url = 'core/actions'
+    basename = 'actions'
+    serializer_class = ActionSerializer
+    pagination_class = HistoryResultsSetPagination
+
+    def get_queryset(self):
+        qs = Action.objects.filter(data__instance_id=self.instance.id)
+        if self.request.user.is_superuser:
+            return qs
+        user_role = self.request.user.get_role(self.instance)
+        if user_role.is_owner:
+            return qs
+        Action.objects.none()
+
+
 class SettingsViewSet(InstanceMixin, viewsets.GenericViewSet):
     url = 'core/settings'
     basename = 'settings'
@@ -527,6 +545,7 @@ class SettingsViewSet(InstanceMixin, viewsets.GenericViewSet):
             main_alarm_group_id = main_alarm_group.id
 
         return RESTResponse({
+            'hub_uid': dynamic_settings['core__hub_uid'],
             'instance_name': self.instance.name,
             'instance_uid': self.instance.uid,
             'timezone': self.instance.timezone,
@@ -626,7 +645,8 @@ class ControllerTypes(InstanceMixin, viewsets.GenericViewSet):
                 'name': cls.name,
                 'is_discoverable': cls.is_discoverable,
                 'manual_add': cls.manual_add,
-                'discovery_msg': cls.discovery_msg
+                'discovery_msg': cls.discovery_msg,
+                'info': cls.info(cls)
             })
 
         return RESTResponse(data)

simo/core/api_meta.py

@@ -1,8 +1,10 @@
 from collections import OrderedDict
+from django.urls import reverse
 from django.utils.encoding import force_str
-from rest_framework import serializers
 from rest_framework.metadata import SimpleMetadata
+from rest_framework import serializers
 from rest_framework.utils.field_mapping import ClassLookupDict
+from simo.core.models import Icon
 from .serializers import (
     HiddenSerializerField, ComponentManyToManyRelatedField,
     TextAreaSerializerField
@@ -40,7 +42,9 @@ class SIMOAPIMetadata(SimpleMetadata):
         TextAreaSerializerField: 'textarea',
     })
 
+
     def get_field_info(self, field):
+
         """
         Given an instance of a serializer field, return a dictionary
         of metadata about it.
@@ -51,6 +55,7 @@ class SIMOAPIMetadata(SimpleMetadata):
 
         form_field = field.style.get('form_field')
         if form_field:
+            #TODO: Delete these completely once autocomplete fields are fully implemented
             if hasattr(form_field, 'queryset'):
                 model = form_field.queryset.model
                 field_info['related_object'] = ".".join(
@@ -59,6 +64,10 @@ class SIMOAPIMetadata(SimpleMetadata):
             if hasattr(form_field, 'filter_by'):
                 field_info['filter_by'] = form_field.filter_by
 
+            if hasattr(form_field, 'forward'):
+                field_info['autocomplete_url'] = reverse(form_field.url)
+                field_info['forward'] = form_field.forward
+
         attrs = [
             'read_only', 'label', 'help_text',
             'min_length', 'max_length',
@@ -76,17 +85,18 @@ class SIMOAPIMetadata(SimpleMetadata):
         elif getattr(field, 'fields', None):
             field_info['children'] = self.get_serializer_info(field)
 
+        if form_field and hasattr(form_field, 'queryset') \
+        and form_field.queryset.model == Icon:
+            return field_info
+
         if not field_info.get('read_only') and hasattr(field, 'choices'):
-            add_choices = True
-            queryset = getattr(field, 'queryset', None)
-            if queryset and queryset.count() > 1000:
-                add_choices = False
-            if add_choices:
-                field_info['choices'] = [
-                    {
-                        'value': choice_value,
-                        'display_name': force_str(choice_name, strings_only=True)
-                    }
-                    for choice_value, choice_name in field.choices.items()
-                ]
+            print(f"DO choices for: {field.label}")
+            field_info['choices'] = [
+                {
+                    'value': choice_value,
+                    'display_name': force_str(choice_name, strings_only=True)
+                }
+                for choice_value, choice_name in field.choices.items()
+            ]
+
         return field_info

simo/core/app_widgets.py

@@ -23,6 +23,12 @@ class BinarySensorWidget(BaseAppWidget):
     size = [2, 1]
 
 
+class ButtonWidget(BaseAppWidget):
+    uid = 'button'
+    name = _("Button")
+    size = [2, 1]
+
+
 class NumericSensorWidget(BaseAppWidget):
     uid = 'numeric-sensor'
     name = _("Numeric sensor")

simo/core/apps.py

@@ -0,0 +1,13 @@
+from django.apps import AppConfig
+
+
+
+class SIMOCoreAppConfig(AppConfig):
+    name = 'simo.core'
+
+    def ready(self):
+        from actstream import registry
+        registry.register(self.get_model('Component'))
+        registry.register(self.get_model('Gateway'))
+
+

simo/core/auto_urls.py

@@ -1,9 +1,9 @@
 from django.urls import path
 from .views import (
-    get_timestamp, setup_wizard, update, restart, reboot, set_instance
+    get_timestamp, update, restart, reboot, set_instance
 )
 from .autocomplete_views import (
-    IconModelAutocomplete, #IconSlugAutocomplete,
+    IconModelAutocomplete,
     CategoryAutocomplete, ZoneAutocomplete,
     ComponentAutocomplete,
 )
@@ -31,7 +31,6 @@ urlpatterns = [
         ComponentAutocomplete.as_view(), name='autocomplete-component'
     ),
     path('set-instance/<slug:instance_slug>/', set_instance, name='set-instance'),
-    path('setup/', setup_wizard, name='setup-wizard'),
     path('update/', update, name='update'),
     path('restart/', restart, name='restart'),
     path('reboot/', reboot, name='reboot')

simo/core/base_types.py

@@ -8,6 +8,7 @@ BASE_TYPES = {
     'numeric-sensor': _("Numeric sensor"),
     'multi-sensor': _("Multi sensor"),
     'binary-sensor': _("Binary sensor"),
+    'button': _("Button"),
     'dimmer': _("Dimmer"),
     'dimmer-plus': _("Dimmer Plus"),
     'rgbw-light': _('RGB(W) light'),

simo/core/controllers.py

@@ -5,8 +5,10 @@ import statistics
 import threading
 from decimal import Decimal as D
 from abc import ABC, ABCMeta, abstractmethod
+from django.utils.functional import cached_property
 from django.core.exceptions import ValidationError
 from django.utils import timezone
+from django.template.loader import render_to_string
 from django.utils.translation import gettext_lazy as _
 from simo.users.middleware import introduce, get_current_user
 from simo.users.utils import get_device_user
@@ -73,6 +75,11 @@ class ControllerBase(ABC):
         :return: Default value of this base component type
         """
 
+    @property
+    def info_template_path(self) -> str:
+        return f"{self.__class__.__module__.split('.')[-2]}/" \
+               f"controllers_info/{self.__class__.__name__.lower()}.md"
+
     @abstractmethod
     def _validate_val(self, value, occasion=None):
         """
@@ -113,6 +120,24 @@ class ControllerBase(ABC):
             cls, '_process_discovery'
         )
 
+    def info(self):
+        '''
+        Override this to give users help on how to use this component type,
+        after you do that, include any component instance specific information
+        if you see it necessary.
+        :return: Markdown component info on how to set it up and use it
+        along with any other relative information,
+         regarding this particular component instance
+        '''
+        try:
+            return render_to_string(
+                self.info_template_path, {
+                    'component': self.component if hasattr(self, 'component') else None
+                }
+            )
+        except:
+            return
+
     def _aggregate_values(self, values):
         if type(values[0]) in (float, int):
             return [statistics.mean(values)]
@@ -306,6 +331,8 @@ class ControllerBase(ABC):
         return value
 
 
+
+
 class TimerMixin:
 
     def __init__(self, *args, **kwargs):
@@ -459,6 +486,36 @@ class BinarySensor(ControllerBase):
         return value
 
 
+class Button(ControllerBase):
+    name = _("Button")
+    base_type = 'button'
+    app_widget = ButtonWidget
+    admin_widget_template = 'admin/controller_widgets/button.html'
+    default_value = 'up'
+
+    def _validate_val(self, value, occasion=None):
+        if value not in ('down', 'up', 'hold', 'click', 'double-click'):
+            raise ValidationError("Bad button value!")
+        return value
+
+    def is_down(self):
+        return self.component.value in ('down', 'hold')
+
+    def is_held(self):
+        return self.component.value == 'hold'
+
+    @cached_property
+    def bonded_gear(self):
+        from simo.core.models import Component
+        gear = []
+        for comp in Component.objects.filter(config__has_key='controls'):
+            for ctrl in comp.config['controls']:
+                if ctrl.get('button') == self.component.id:
+                    gear.append(comp)
+                    break
+        return gear
+
+
 class OnOffPokerMixin:
     _poke_toggle = False
 

simo/core/dynamic_settings.py

@@ -43,14 +43,6 @@ class RemoteConnectionVersion(IntegerPreference):
                 "when hub get's synced up to the simo.io."
 
 
-@global_preferences_registry.register
-class UnitsOfMeasure(ChoicePreference):
-    section = core
-    name = 'units_of_measure'
-    choices = (('metric', "Metric"), ('imperial', "Imperial"))
-    default = 'metric'
-    required = True
-
 
 @global_preferences_registry.register
 class LatestHubOSVersionAvailable(StringPreference):

simo/core/form_fields.py

@@ -0,0 +1,93 @@
+import copy
+from django import forms
+from dal import autocomplete
+
+
+class LazyChoicesMixin:
+
+    _choices = ()
+
+    @property
+    def choices(self):
+        if callable(self._choices):
+            return self._choices()
+        return self._choices
+
+    @choices.setter
+    def choices(self, value):
+        self._choices = value
+
+    def __deepcopy__(self, memo):
+        obj = copy.copy(self)
+        obj.attrs = self.attrs.copy()
+        if not callable(self._choices):
+            obj._choices = copy.copy(self._choices)
+        memo[id(self)] = obj
+        return obj
+
+    def optgroups(self, name, value, attrs=None):
+        for (val, display) in self.choices:
+            if val == value[0]:
+                self.choices = [(val, display)]
+                break
+        return super().optgroups(name, value, attrs)
+
+
+class ListSelect2(LazyChoicesMixin, autocomplete.ListSelect2):
+    pass
+
+
+class Select2Multiple(LazyChoicesMixin, autocomplete.Select2Multiple):
+    pass
+
+
+class Select2ListMixin:
+
+    def __init__(self, url, forward=None, *args, **kwargs):
+        self.url = url
+        self.forward = []
+        if forward:
+            self.forward = [fw.to_dict() for fw in forward]
+
+        widget = ListSelect2(
+            url=url, forward=forward, attrs={'data-html': True},
+
+        )
+        widget.choices = kwargs.get('choices', None)
+
+        super().__init__(widget=widget, *args, **kwargs)
+
+class Select2ModelChoiceField(Select2ListMixin, forms.ModelChoiceField):
+    pass
+
+
+class Select2ListChoiceField(Select2ListMixin, forms.ChoiceField):
+    pass
+
+
+class Select2MultipleMixin:
+
+    def __init__(self, url, forward=None, *args, **kwargs):
+        self.url = url
+        self.forward = []
+        if forward:
+            self.forward = [fw.to_dict() for fw in forward]
+
+        widget = Select2Multiple(
+            url=url, forward=forward, attrs={'data-html': True}
+        )
+        widget.choices = kwargs.pop('choices', None)
+
+        super().__init__(widget, *args, **kwargs)
+
+
+class Select2ModelMultipleChoiceField(
+    Select2MultipleMixin, forms.ModelMultipleChoiceField
+):
+    pass
+
+
+class Select2ListMultipleChoiceField(
+    Select2MultipleMixin, forms.MultipleChoiceField
+):
+    pass