Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 1 | from rest_framework.decorators import api_view |
| 2 | from rest_framework.response import Response |
| 3 | from rest_framework.reverse import reverse |
| 4 | from rest_framework import serializers |
| 5 | from rest_framework import generics |
| 6 | from rest_framework import viewsets |
| 7 | from rest_framework.decorators import detail_route, list_route |
| 8 | from rest_framework.views import APIView |
| 9 | from core.models import * |
| 10 | from django.forms import widgets |
| 11 | from django.conf.urls import patterns, url |
| 12 | from services.cord.models import VOLTTenant, VBNGTenant, CordSubscriberRoot |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 13 | from api.xosapi_helpers import PlusSerializerMixin, XOSViewSet, ReadOnlyField |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 14 | from django.shortcuts import get_object_or_404 |
| 15 | from xos.apibase import XOSListCreateAPIView, XOSRetrieveUpdateDestroyAPIView, XOSPermissionDenied |
| 16 | from xos.exceptions import * |
| 17 | import json |
| 18 | import subprocess |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 19 | from django.views.decorators.csrf import ensure_csrf_cookie |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 20 | |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 21 | class CordSubscriberNew(CordSubscriberRoot): |
| 22 | class Meta: |
| 23 | proxy = True |
| 24 | app_label = "cord" |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 25 | |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 26 | def __init__(self, *args, **kwargs): |
| 27 | super(CordSubscriberNew, self).__init__(*args, **kwargs) |
| 28 | |
| 29 | def __unicode__(self): |
| 30 | return u"cordSubscriber-%s" % str(self.id) |
| 31 | |
| 32 | @property |
| 33 | def features(self): |
| 34 | return {"cdn": self.cdn_enable, |
| 35 | "uplink_speed": self.uplink_speed, |
| 36 | "downlink_speed": self.downlink_speed, |
| 37 | "uverse": self.enable_uverse, |
| 38 | "status": self.status} |
| 39 | |
| 40 | @features.setter |
| 41 | def features(self, value): |
| 42 | self.cdn_enable = value.get("cdn", self.get_default_attribute("cdn_enable")) |
| 43 | self.uplink_speed = value.get("uplink_speed", self.get_default_attribute("uplink_speed")) |
| 44 | self.downlink_speed = value.get("downlink_speed", self.get_default_attribute("downlink_speed")) |
| 45 | self.enable_uverse = value.get("uverse", self.get_default_attribute("enable_uverse")) |
| 46 | self.status = value.get("status", self.get_default_attribute("status")) |
| 47 | |
Scott Baker | c4200f9 | 2016-03-28 15:07:36 -0700 | [diff] [blame] | 48 | def update_features(self, value): |
| 49 | d=self.features |
| 50 | d.update(value) |
| 51 | self.features = d |
| 52 | |
Scott Baker | 85f2cd3 | 2016-03-28 16:23:53 -0700 | [diff] [blame^] | 53 | @property |
| 54 | def identity(self): |
| 55 | return {"account_num": self.service_specific_id} |
| 56 | |
| 57 | @identity.setter |
| 58 | def identity(self, value): |
| 59 | self.service_specific_id = value.get("account_num", "") |
| 60 | |
| 61 | def update_identity(self, value): |
| 62 | d=self.identity |
| 63 | d.update(value) |
| 64 | self.identity = d |
| 65 | |
| 66 | @property |
| 67 | def related(self): |
| 68 | related = {} |
| 69 | if self.volt: |
| 70 | related["volt_id"] = self.volt.id |
| 71 | related["s_tag"] = self.volt.s_tag |
| 72 | related["c_tag"] = self.volt.c_tag |
| 73 | if self.volt.vcpe: |
| 74 | related["vsg_id"] = self.volt.vcpe.id |
| 75 | if self.volt.vcpe.instance: |
| 76 | related["instance_id"] = self.volt.vcpe.instance.id |
| 77 | related["instance_name"] = self.volt.vcpe.instance.name |
| 78 | related["wan_container_ip"] = self.volt.vcpe.wan_container_ip |
| 79 | return related |
| 80 | |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 81 | def save(self, *args, **kwargs): |
| 82 | super(CordSubscriberNew, self).save(*args, **kwargs) |
| 83 | |
Scott Baker | c4200f9 | 2016-03-28 15:07:36 -0700 | [diff] [blame] | 84 | class FeatureSerializer(serializers.Serializer): |
| 85 | cdn = serializers.BooleanField(required=False) |
| 86 | uplink_speed = serializers.IntegerField(required=False) |
| 87 | downlink_speed = serializers.IntegerField(required=False) |
| 88 | uverse = serializers.BooleanField(required=False) |
| 89 | status = serializers.CharField(required=False) |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 90 | |
Scott Baker | 85f2cd3 | 2016-03-28 16:23:53 -0700 | [diff] [blame^] | 91 | class IdentitySerializer(serializers.Serializer): |
| 92 | account_num = serializers.CharField(required=False) |
| 93 | |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 94 | class CordSubscriberSerializer(serializers.ModelSerializer, PlusSerializerMixin): |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 95 | id = ReadOnlyField() |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 96 | humanReadableName = serializers.SerializerMethodField("getHumanReadableName") |
Scott Baker | 85f2cd3 | 2016-03-28 16:23:53 -0700 | [diff] [blame^] | 97 | features = FeatureSerializer() |
| 98 | identity = IdentitySerializer() |
| 99 | related = serializers.DictField(required=False) |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 100 | |
| 101 | class Meta: |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 102 | model = CordSubscriberNew |
| 103 | fields = ('humanReadableName', |
| 104 | 'id', |
Scott Baker | 85f2cd3 | 2016-03-28 16:23:53 -0700 | [diff] [blame^] | 105 | 'features', |
| 106 | 'identity', |
| 107 | 'related') |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 108 | |
| 109 | def getHumanReadableName(self, obj): |
| 110 | return obj.__unicode__() |
| 111 | |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 112 | # @ensure_csrf_cookie |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 113 | class CordSubscriberViewSet(XOSViewSet): |
| 114 | base_name = "subscriber" |
Scott Baker | d9d55f2 | 2016-03-25 13:33:11 -0700 | [diff] [blame] | 115 | method_name = "subscriber" |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 116 | method_kind = "viewset" |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 117 | queryset = CordSubscriberNew.get_tenant_objects().select_related().all() |
| 118 | serializer_class = CordSubscriberSerializer |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 119 | |
| 120 | @classmethod |
Scott Baker | bca3a1b | 2016-03-28 13:22:10 -0700 | [diff] [blame] | 121 | def get_urlpatterns(self, api_path="^"): |
Scott Baker | d9d55f2 | 2016-03-25 13:33:11 -0700 | [diff] [blame] | 122 | patterns = super(CordSubscriberViewSet, self).get_urlpatterns(api_path=api_path) |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 123 | patterns.append( self.detail_url("features/$", {"get": "get_features", "put": "set_features"}, "features") ) |
| 124 | patterns.append( self.detail_url("features/(?P<feature>[a-zA-Z0-9\-_]+)/$", {"get": "get_feature", "put": "set_feature"}, "get_feature") ) |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 125 | |
Scott Baker | d9d55f2 | 2016-03-25 13:33:11 -0700 | [diff] [blame] | 126 | patterns.append( url(self.api_path + "subidlookup/(?P<ssid>[0-9\-]+)/$", self.as_view({"get": "ssiddetail"}), name="ssiddetail") ) |
| 127 | patterns.append( url(self.api_path + "subidlookup/$", self.as_view({"get": "ssidlist"}), name="ssidlist") ) |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 128 | |
| 129 | return patterns |
| 130 | |
| 131 | def list(self, request): |
| 132 | object_list = self.filter_queryset(self.get_queryset()) |
| 133 | |
| 134 | serializer = self.get_serializer(object_list, many=True) |
| 135 | |
Scott Baker | c4200f9 | 2016-03-28 15:07:36 -0700 | [diff] [blame] | 136 | return Response(serializer.data) |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 137 | |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 138 | def get_features(self, request, pk=None): |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 139 | subscriber = self.get_object() |
Scott Baker | c4200f9 | 2016-03-28 15:07:36 -0700 | [diff] [blame] | 140 | return Response(FeatureSerializer(subscriber.features).data) |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 141 | |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 142 | def get_feature(self, request, pk=None, feature=None): |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 143 | subscriber = self.get_object() |
Scott Baker | c4200f9 | 2016-03-28 15:07:36 -0700 | [diff] [blame] | 144 | return Response({feature: FeatureSerializer(subscriber.features).data[feature]}) |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 145 | |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 146 | def set_feature(self, request, pk=None, feature=None): |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 147 | subscriber = self.get_object() |
Scott Baker | c4200f9 | 2016-03-28 15:07:36 -0700 | [diff] [blame] | 148 | if [feature] != request.data.keys(): |
| 149 | raise serializers.ValidationError("feature %s does not match keys in request body (%s)" % (feature, ",".join(request.data.keys()))) |
| 150 | ser = FeatureSerializer(subscriber.features, data=request.data) |
| 151 | ser.is_valid(raise_exception = True) |
| 152 | subscriber.update_features(ser.validated_data) |
Scott Baker | 85f2cd3 | 2016-03-28 16:23:53 -0700 | [diff] [blame^] | 153 | return Response({feature: FeatureSerializer(subscriber.features).data[feature]}) |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 154 | |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 155 | def set_features(self, request, pk=None): |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 156 | subscriber = self.get_object() |
Scott Baker | c4200f9 | 2016-03-28 15:07:36 -0700 | [diff] [blame] | 157 | ser = FeatureSerializer(subscriber.features, data=request.data) |
| 158 | ser.is_valid(raise_exception = True) |
| 159 | subscriber.update_features(ser.validated_data) |
| 160 | subscriber.save() |
| 161 | return Response(FeatureSerializer(subscriber.features).data) |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 162 | |
| 163 | def ssidlist(self, request): |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 164 | object_list = CordSubscriberNew.get_tenant_objects().all() |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 165 | |
| 166 | ssidmap = [ {"service_specific_id": x.service_specific_id, "subscriber_id": x.id} for x in object_list ] |
| 167 | |
| 168 | return Response({"ssidmap": ssidmap}) |
| 169 | |
| 170 | def ssiddetail(self, pk=None, ssid=None): |
Scott Baker | 12d7108 | 2016-03-28 13:23:20 -0700 | [diff] [blame] | 171 | object_list = CordSubscriberNew.get_tenant_objects().all() |
Scott Baker | fb9544a | 2016-03-25 10:55:03 -0700 | [diff] [blame] | 172 | |
| 173 | ssidmap = [ {"service_specific_id": x.service_specific_id, "subscriber_id": x.id} for x in object_list if str(x.service_specific_id)==str(ssid) ] |
| 174 | |
| 175 | if len(ssidmap)==0: |
| 176 | raise XOSNotFound("didn't find ssid %s" % str(ssid)) |
| 177 | |
| 178 | return Response( ssidmap[0] ) |
| 179 | |