1. @login_required()
2. def list(request):
3.     services = Service.objects.annotate(incident_num=Count('incident'))
4.     return TemplateResponse(request, 'services/list.html', {'services': services})
5.  
6. @login_required()
7. def delete(request, id):
8.     try:
9.         service = Service.objects.get(id = id)
10.         check_user_ownership(request, service)
11.         service.delete()
12.         return HttpResponseRedirect('/services/')
13.     except Service.DoesNotExist:
14.         raise Http404
15.  
16. @login_required()
17. def edit(request, id):
18.     try:
19.         service = Service.objects.get(id = id)
20.         try:
21.             api_keys = ServiceTokens.objects.filter(service_id = service)
22.         except ServiceTokens.DoesNotExist:
23.             api_keys = []
24.         try:
25.             ss_obj = ServiceSilenced.objects.get(service=service)
26.             service_silenced = str(ss_obj.silenced_until - timezone.now()).split(".")[0]
27.         except ServiceSilenced.DoesNotExist:
28.             service_silenced = False
29.  
30.         # get the list of users
31.         users = User.objects.all()
32.  
33.         policy = service.policy if service.policy else None
34.         all_polcies = SchedulePolicy.objects.all()
35.         return TemplateResponse(request, 'services/edit.html', {
36.             'item': service,
37.             'policy': policy,
38.             'policies': all_polcies,
39.             'api_keys': api_keys,
40.             'service_silenced': service_silenced,
41.             'url': request.get_full_path(),
42.             'users': users
43.         })
44.     except Service.DoesNotExist:
45.         raise Http404
46.  
47. @login_required()
48. def new(request):
49.     all_policies = SchedulePolicy.objects.all()
50.     # get the list of users
51.     users = [request.user]
52.  
53.     return TemplateResponse(request, 'services/edit.html', {'policies': all_policies,
54.                                                             'users': users})
55.  
56. @login_required()
57. @require_http_methods(["POST"])
58. def save(request):
59.     # Update service fields
60.     creation_time = False
61.     try:
62.         service_id = request.POST.get("id")
63.         if service_id != '0':
64.             service = Service.objects.get(id=service_id)
65.             check_user_ownership(request, service)
66.  
67.         else:
68.             creation_time = True
69.             service = Service()
70.     except Service.DoesNotExist:
71.         creation_time = True
72.         service = Service()
73.  
74.     if 'owner' in request.POST:
75.         service.owner = User.objects.get(id=request.POST['owner'])
76.     else:
77.         service.owner = request.user
78.  
79.  
80.     service.name = request.POST['name']
81.     if len(service.name) < 6:
82.         messages.error(request, "The Service Name should be at least 6 characters!")
83.         if request.META.get('HTTP_REFERER').endswith("new"):
84.             # if the user was making a new service
85.             return HttpResponseRedirect(reverse('openduty.services.new'))
86.         else:
87.             # if the user editing existing page
88.             return HttpResponseRedirect(reverse('openduty.services.edit', None, [str(request.POST['id'])]))
89.  
90.     service.escalate_after = request.POST['escalate_after']
91.     service.retry = request.POST['retry']
92.     service.notifications_disabled = request.POST.get("disable_notification", "off") == "on"
93.     if(request.POST['policy']):
94.         pol = SchedulePolicy.objects.get(id = request.POST['policy'])
95.     else:
96.         pol = None
97.     service.policy = pol
98.     # Save service
99.     try:
100.         service.save()
101.         if creation_time:
102.             # Auto creating an email endpoint
103.             valid_email_chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789!#$%&'*+-/=?^_`{|}~."
104.             email = "".join(c for c in request.POST['name'] if c in valid_email_chars) + "@" + settings.HOSTNAME
105.             email_token_create(email, service.id)
106.     except IntegrityError:
107.         messages.error(request, 'Service validation failed')
108.         if len (request.POST['id']) > 0:
109.             return HttpResponseRedirect(reverse('openduty.services.edit', None, [str(request.POST['id'])]))
110.         else:
111.             return HttpResponseRedirect(reverse('openduty.services.new'))
112.  
113.     return HttpResponseRedirect('/services/')
114.  
115. @login_required()
116. def token_delete(request, token_id):
117.     try:
118.         token = ServiceTokens.objects.get(id = token_id)
119.         token.delete()
120.         return HttpResponseRedirect(reverse('openduty.services.edit', None, [str(token.service_id.id)]));
121.     except Service.DoesNotExist:
122.         raise Http404
123.  
124.  
125. def email_token_create(email_address, service_id):
126.     service = Service.objects.get(id=service_id)
127.     token = Token()
128.     token.save_email(email_address=email_address)
129.     service_token = ServiceTokens.objects.create(service_id=service, token_id=token, name="Email Integration")
130.     service_token.save()
131.  
132. @login_required()
133. @require_http_methods(["POST"])
134. def token_create(request, service_id):
135.     try:
136.         service = Service.objects.get(id = service_id)
137.         check_user_ownership(request, service)
138.         token = Token()
139.         token.save()
140.         service_token = ServiceTokens.objects.create(service_id = service, token_id = token, name=request.POST['key_name'])
141.         service_token.save()
142.         return HttpResponseRedirect(reverse('openduty.services.edit', None, [str(service_id)]));
143.     except Service.DoesNotExist:
144.         raise Http404
145.  
146. @login_required()
147. @require_http_methods(["POST"])
148. def silence(request, service_id):
149.     try:
150.         service = Service.objects.get(id = service_id)
151.         silence_for = request.POST.get('silence_for')
152.         url = request.POST.get("url")
153.         incident_id = request.POST.get('incident_id')
154.         if incident_id:
155.             incident = Incident.objects.get(id=incident_id)
156.         else:
157.             incident = None
158.         if ServiceSilenced.objects.filter(service=service).count() < 1:
159.             silenced_service = ServiceSilenced()
160.             silenced_service.service = service
161.             silenced_service.silenced_until = timezone.now() + timezone.timedelta(hours=int(silence_for))
162.             silenced_service.silenced = True
163.             silenced_service.save()
164.  
165.             event_log_message = "%s silenced the service %s for %s hours" % (request.user.username, service.name, silence_for)
166.             event_log = EventLog()
167.             event_log.incident_key = incident
168.             event_log.action = 'silence_service'
169.             event_log.user = request.user
170.             event_log.service_key = service
171.             event_log.data = event_log_message
172.             event_log.occurred_at = timezone.now()
173.             event_log.save()
174.  
175.             unsilence_service.apply_async((service_id,), eta=silenced_service.silenced_until)
176.         return HttpResponseRedirect(url)
177.     except Service.DoesNotExist:
178.         raise Http404
179.  
180. @login_required()
181. @require_http_methods(["POST"])
182. def unsilence(request, service_id):
183.     try:
184.         service = Service.objects.get(id = service_id)
185.         url = request.POST.get("url")
186.         try:
187.             ServiceSilenced.objects.filter(service=service).delete()
188.             event_log_message = "%s removed silence from service %s" % (request.user.username, service.name)
189.             event_log = EventLog()
190.             event_log.action = 'unsilence_service'
191.             event_log.user = request.user
192.             event_log.incident_key = None
193.             event_log.service_key = service
194.             event_log.data = event_log_message
195.             event_log.occurred_at = timezone.now()
196.             event_log.save()
197.         except ServiceSilenced.DoesNotExist:
198.             # No need to delete
199.             pass
200.         return HttpResponseRedirect(url)
201.     except Service.DoesNotExist:
202.