admin.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877
  1. from django.contrib import admin
  2. from django.db import models
  3. from django.db.models import Q
  4. from django.forms import ModelForm, BaseInlineFormSet
  5. from django.utils import timezone
  6. from django.urls import reverse, path
  7. from django.utils.html import format_html
  8. from django.conf.urls import url
  9. from django.template.response import TemplateResponse
  10. from django.core.serializers import serialize
  11. from django.http import HttpResponse
  12. from django.db.models.functions import Cast
  13. from django.contrib.postgres.aggregates import StringAgg
  14. from django.db import connection, transaction
  15. from django.core.cache import cache
  16. from django.contrib.humanize.templatetags.humanize import naturaltime
  17. from django.contrib.contenttypes.models import ContentType
  18. from django.http import HttpResponseRedirect
  19. from djgeojson.views import GeoJSONLayerView
  20. from urllib.parse import urlencode
  21. from functools import partial, update_wrapper
  22. from datetime import timedelta
  23. from djadhere.utils import get_active_filter
  24. from adhesions.models import Adhesion
  25. from banking.models import PaymentUpdate
  26. from .models import Service, ServiceType, IPPrefix, IPResource, IPResourceState, \
  27. ServiceAllocation, Antenna, AntennaAllocation, Allocation, \
  28. Route, Tunnel, Switch, Port
  29. from .utils.notifications import notify_allocation
  30. from .forms import AntennaForm, StopAllocationForm
  31. ### Filters
  32. class ResourceInUseFilter(admin.SimpleListFilter):
  33. title = 'disponibilité'
  34. parameter_name = 'available'
  35. def lookups(self, request, model_admin):
  36. return (
  37. (1, 'Disponible'),
  38. (0, 'Non disponible'),
  39. )
  40. def queryset(self, request, queryset):
  41. available_filter = Q(reserved=False, in_use=False)
  42. if self.value() == '0': # non disponible
  43. return queryset.exclude(available_filter)
  44. if self.value() == '1': # disponible
  45. return queryset.filter(available_filter)
  46. class ResourcePingFilter(admin.SimpleListFilter):
  47. title = 'ping'
  48. parameter_name = 'ping'
  49. def lookups(self, request, model_admin):
  50. return (
  51. ('up', 'UP'),
  52. ('down', 'DOWN'),
  53. ('down-since', 'DOWN depuis…'),
  54. ('never-up', 'Jamais vu UP'),
  55. )
  56. def queryset(self, request, queryset):
  57. if self.value() == 'up':
  58. return queryset.filter(last_state__state=IPResourceState.STATE_UP)
  59. if self.value() == 'down':
  60. return queryset.exclude(last_state__state=IPResourceState.STATE_UP) # DOWN + UNKNOWN
  61. if self.value() == 'down-since':
  62. queryset = queryset.exclude(last_state__state=IPResourceState.STATE_UP)
  63. return queryset.filter(last_time_up__isnull=False)
  64. if self.value() == 'never-up':
  65. queryset = queryset.exclude(last_state__state=IPResourceState.STATE_UP) # DOWN + UNKWON
  66. return queryset.filter(last_time_up__isnull=True)
  67. class ActiveServiceFilter(admin.SimpleListFilter):
  68. title = 'actif'
  69. parameter_name = 'active'
  70. def lookups(self, request, model_admin):
  71. return (
  72. (1, 'Actif'),
  73. (0, 'Inactif'),
  74. )
  75. def queryset(self, request, queryset):
  76. if self.value() == '0': # inactif
  77. return queryset.exclude(get_active_filter('allocation'))
  78. if self.value() == '1': # actif
  79. return queryset.filter(get_active_filter('allocation'))
  80. class RouteFilter(admin.SimpleListFilter):
  81. title = 'route'
  82. parameter_name = 'route'
  83. def lookups(self, request, model_admin):
  84. return ServiceAllocation.objects.filter(active=True).values_list('route__pk', 'route__name').distinct()
  85. def queryset(self, request, queryset):
  86. try:
  87. route = int(self.value())
  88. except (TypeError, ValueError):
  89. pass
  90. else:
  91. allocations = ServiceAllocation.objects.filter(active=True, route__pk=route).values_list('pk', flat=True)
  92. queryset = queryset.filter(service_allocation__in=allocations)
  93. return queryset
  94. class AntennaPrefixFilter(admin.SimpleListFilter):
  95. title = 'préfix'
  96. parameter_name = 'prefix'
  97. def lookups(self, request, model_admin):
  98. resources = AntennaAllocation.objects.filter(active=True).values_list('resource__pk', flat=True)
  99. prefixes = IPPrefix.objects.filter(ipresource__in=resources).values_list('pk', 'prefix').distinct()
  100. return prefixes
  101. def queryset(self, request, queryset):
  102. try:
  103. prefix = int(self.value())
  104. except (TypeError, ValueError):
  105. pass
  106. else:
  107. allocations = AntennaAllocation.objects.filter(active=True, resource__prefixes__pk=prefix).values_list('pk', flat=True)
  108. queryset = queryset.filter(allocation__in=allocations)
  109. return queryset
  110. class AntennaPositionFilter(admin.SimpleListFilter):
  111. title = 'géolocalisation'
  112. parameter_name = 'position'
  113. def lookups(self, request, model_admin):
  114. return (
  115. ('1', 'Connue'),
  116. ('0', 'Inconnue'),
  117. )
  118. def queryset(self, request, queryset):
  119. query = Q(position__isnull=True)
  120. if self.value() == '0':
  121. return queryset.filter(query)
  122. if self.value() == '1':
  123. return queryset.exclude(query)
  124. return queryset
  125. class ActiveTunnelFilter(admin.SimpleListFilter):
  126. title = 'status'
  127. parameter_name = 'active'
  128. def lookups(self, request, model_admin):
  129. return (
  130. ('1', 'Actif'),
  131. ('0', 'Désactivé'),
  132. )
  133. def queryset(self, request, queryset):
  134. query = Q(ended__isnull=True)
  135. if self.value() == '0':
  136. return queryset.exclude(query)
  137. if self.value() == '1':
  138. return queryset.filter(query)
  139. return queryset
  140. ### Inlines
  141. class AllocationInlineFormSet(BaseInlineFormSet):
  142. def save_new(self, form, commit=True):
  143. obj = super().save_new(form, commit)
  144. if type(obj) == ServiceAllocation:
  145. notify_allocation(self.request, obj)
  146. return obj
  147. def save_existing(self, form, instance, commit=True):
  148. old = type(instance).objects.get(pk=instance.pk)
  149. if type(instance) == ServiceAllocation:
  150. notify_allocation(self.request, instance, old)
  151. return super().save_existing(form, instance, commit)
  152. class AllocationInline(admin.TabularInline):
  153. formset = AllocationInlineFormSet
  154. extra = 0
  155. show_change_link = True
  156. ordering = ('-start',)
  157. def resource_link(self, obj):
  158. url = reverse('admin:services_ipresource_change', args=[obj.resource.pk])
  159. return format_html('<a href="{}">{}</a>', url, str(obj.resource))
  160. resource_link.short_description = 'IP'
  161. def get_queryset(self, request):
  162. qs = super().get_queryset(request)
  163. qs = qs.select_related('resource')
  164. return qs
  165. def get_formset(self, request, obj=None, **kwargs):
  166. formset = super().get_formset(request, obj, **kwargs)
  167. formset.request = request
  168. return formset
  169. def has_delete_permission(self, request, obj=None):
  170. return False
  171. class NewAllocationMixin:
  172. verbose_name_plural = 'Nouvelle allocation'
  173. max_num = 1
  174. def get_queryset(self, request):
  175. return super().get_queryset(request).model.objects.none()
  176. class ActiveAllocationMixin:
  177. verbose_name_plural = 'Allocations actives'
  178. max_num = 0
  179. def get_queryset(self, request):
  180. return super().get_queryset(request).filter(get_active_filter())
  181. class InactiveAllocationMixin:
  182. verbose_name_plural = 'Anciennes allocations'
  183. max_num = 0
  184. def get_queryset(self, request):
  185. return super().get_queryset(request).exclude(get_active_filter())
  186. class ServiceAllocationMixin:
  187. model = ServiceAllocation
  188. fields = ('id', 'service', 'resource', 'route', 'start', 'end')
  189. raw_id_fields = ('resource',)
  190. autocomplete_fields = ('service',)
  191. def service_link(self, obj):
  192. url = reverse('admin:services_service_change', args=[obj.service.pk])
  193. return format_html('<a href="{}">{}</a>', url, str(obj.service))
  194. service_link.short_description = 'Service'
  195. def get_queryset(self, request):
  196. qs = super().get_queryset(request)
  197. qs = qs.select_related('route')
  198. return qs
  199. #class AntennaAllocationMixin:
  200. # model = AntennaAllocation
  201. # fields = ('id', 'antenna', 'resource', 'start', 'end')
  202. # raw_id_fields = ('resource',)
  203. # autocomplete_fields = ('antenna',)
  204. #
  205. # def get_queryset(self, request):
  206. # qs = super().get_queryset(request)
  207. # qs = qs.select_related('antenna')
  208. # return qs
  209. class NewServiceAllocationInline(ServiceAllocationMixin, NewAllocationMixin, AllocationInline):
  210. fields = ('id', 'service', 'resource', 'route',)
  211. class ActiveServiceAllocationInline(ServiceAllocationMixin, ActiveAllocationMixin, AllocationInline):
  212. fields = ('id', 'service_link', 'resource_link', 'route', 'start', 'stop',)
  213. readonly_fields = ('service_link', 'start', 'resource_link', 'stop',)
  214. def stop(self, obj):
  215. return format_html('<a href="{}" class="deletelink">Terminer</a>', reverse('admin:stop-allocation', kwargs={'resource': obj.resource.ip}))
  216. stop.short_description = 'Terminer l’allocation'
  217. class InactiveServiceAllocationInline(ServiceAllocationMixin, InactiveAllocationMixin, AllocationInline):
  218. fields = ('id', 'service_link', 'resource_link', 'route', 'start', 'end')
  219. readonly_fields = ('service_link', 'resource_link', 'route', 'start', 'end')
  220. #class ActiveAntennaAllocationInline(AntennaAllocationMixin, ActiveAllocationMixin, AllocationInline):
  221. # pass
  222. #class InactiveAntennaAllocationInline(AntennaAllocationMixin, InactiveAllocationMixin, AllocationInline):
  223. # pass
  224. class IPResourceStateInline(admin.TabularInline):
  225. model = IPResourceState
  226. verbose_name_plural = 'Historique des derniers changements d’état'
  227. fields = ['date']
  228. readonly_fields = ['date']
  229. ordering = ['-date']
  230. def has_add_permission(self, request):
  231. return False
  232. def has_delete_permission(self, request, obj=None):
  233. return False
  234. class PortInline(admin.TabularInline):
  235. model = Port
  236. max_num = 0
  237. def has_add_permission(self, request):
  238. return False
  239. def has_delete_permission(self, request, obj=None):
  240. return False
  241. class SwitchPortInline(PortInline):
  242. fields = ('port', 'up', 'reserved', 'service', 'notes',)
  243. readonly_fields = ('port', 'up',)
  244. autocomplete_fields = ('service',)
  245. def get_queryset(self, request):
  246. qs = super().get_queryset(request)
  247. qs = qs.select_related('switch', 'service', 'service__service_type')
  248. return qs
  249. class ServicePortInline(PortInline):
  250. fields = ('switch', 'port', 'up', 'notes',)
  251. readonly_fields = ('switch', 'port', 'up',)
  252. ### ModelAdmin
  253. class ServiceAdmin(admin.ModelAdmin):
  254. list_display = ('id', 'get_adhesion_link', 'get_adherent_link', 'service_type', 'label', 'is_active',)
  255. list_select_related = ('adhesion', 'adhesion__user', 'adhesion__user__profile', 'adhesion__corporation', 'service_type',)
  256. list_filter = (
  257. ActiveServiceFilter,
  258. 'loan_equipment',
  259. ('service_type', admin.RelatedOnlyFieldListFilter),
  260. )
  261. search_fields = ('=id', 'service_type__name', 'label', 'notes',)
  262. fields = ('adhesion', 'service_type', 'label', 'notes', 'loan_equipment', 'get_contribution_link', 'is_active',)
  263. readonly_fields = ('get_contribution_link', 'is_active',)
  264. raw_id_fields = ('adhesion',)
  265. def save_model(self, request, srv, form, change):
  266. if 'adhesion' in form.changed_data:
  267. with transaction.atomic():
  268. old_srv = Service.objects.get(pk=srv.pk)
  269. adhesion = srv.adhesion
  270. srv.adhesion = old_srv.adhesion
  271. label = srv.label
  272. srv.label = '%s (transféré à ADT%d le %s)' % (srv.label, adhesion.pk, timezone.now().strftime('%d/%m/%Y'))
  273. srv.save()
  274. new_srv = Service.objects.create(adhesion=adhesion, service_type=srv.service_type, label=label,
  275. notes=srv.notes, loan_equipment=srv.loan_equipment)
  276. for allocation in srv.active_allocations:
  277. allocation.end = timezone.now()
  278. allocation.save()
  279. ServiceAllocation.objects.create(resource=allocation.resource, service=new_srv, route=allocation.route)
  280. else:
  281. super().save_model(request, srv, form, change)
  282. def get_queryset(self, request):
  283. qs = super().get_queryset(request)
  284. qs = qs.prefetch_related('allocations',)
  285. return qs
  286. get_adhesion_link = lambda self, service: service.adhesion.get_adhesion_link()
  287. get_adhesion_link.short_description = Adhesion.get_adhesion_link.short_description
  288. get_adherent_link = lambda self, service: service.adhesion.get_adherent_link()
  289. get_adherent_link.short_description = Adhesion.get_adherent_link.short_description
  290. def get_contribution_link(self, obj):
  291. return format_html(u'<a href="{}">{}</a>', obj.contribution.get_absolute_url(), obj.contribution.get_current_payment_display())
  292. get_contribution_link.short_description = 'Contribution financière'
  293. def get_inline_instances(self, request, obj=None):
  294. inlines = []
  295. if obj and obj.ports.exists():
  296. inlines += [ServicePortInline]
  297. inlines += [NewServiceAllocationInline]
  298. if obj and obj.active_allocations.exists():
  299. inlines += [ActiveServiceAllocationInline]
  300. if obj and obj.inactive_allocations.exists():
  301. inlines += [InactiveServiceAllocationInline]
  302. return [inline(self.model, self.admin_site) for inline in inlines]
  303. def get_actions(self, request):
  304. actions = super().get_actions(request)
  305. if 'delete_selected' in actions:
  306. del actions['delete_selected']
  307. return actions
  308. def has_delete_permission(self, request, obj=None):
  309. if not obj:
  310. return False
  311. one_year_ago = timezone.now() - timedelta(days=365)
  312. contribution = obj.contribution.updates.filter(validated=True).first()
  313. # s’il y avait un paiement actif il y a moins d’un an
  314. if not contribution or contribution.payment_method != PaymentUpdate.STOP or contribution.start > one_year_ago:
  315. return False
  316. # s’il y avait une allocation active il y a moins d’un an
  317. if any(map(lambda a: a.end is None or a.end > one_year_ago, obj.allocations.all())):
  318. return False
  319. return True
  320. class IPPrefixAdmin(admin.ModelAdmin):
  321. def has_delete_permission(self, request, obj=None):
  322. # Interdiction de supprimer le préfix s’il est assigné à un tunnel
  323. return obj and obj.tunnel_set.exists()
  324. def has_change_permission(self, request, obj=None):
  325. if obj:
  326. return False
  327. else:
  328. return True
  329. # pour embêcher de by-passer le check has_delete_permission, on désactive l’action delete
  330. def get_actions(self, request):
  331. actions = super().get_actions(request)
  332. if 'delete_selected' in actions:
  333. del actions['delete_selected']
  334. return actions
  335. class IPResourceAdmin(admin.ModelAdmin):
  336. list_display = ('__str__', 'available_display', 'route', 'last_use', 'ping',)
  337. list_filter = (
  338. 'category',
  339. ResourceInUseFilter,
  340. ResourcePingFilter,
  341. 'reserved',
  342. ('prefixes', admin.RelatedOnlyFieldListFilter),
  343. RouteFilter,
  344. )
  345. search_fields = ('=ip', 'notes',)
  346. actions = ['contact_ip_owners']
  347. ordering = ['ip']
  348. inlines = [ IPResourceStateInline ]
  349. def get_fields(self, request, obj=None):
  350. return self.get_readonly_fields(request, obj)
  351. def get_readonly_fields(self, request, obj=None):
  352. fields = ['ip']
  353. if obj:
  354. if obj.reserved:
  355. fields += ['reserved']
  356. if not obj.in_use:
  357. fields += ['last_use']
  358. fields += ['last_state']
  359. if obj.last_state.state != IPResourceState.STATE_UP:
  360. fields += ['last_time_up']
  361. if obj.category == IPResource.CATEGORY_PUBLIC:
  362. fields += ['password']
  363. if obj.checkmk_label:
  364. fields += ['checkmk']
  365. if obj.notes:
  366. fields += ['notes']
  367. return fields
  368. def get_inline_instances(self, request, obj=None):
  369. super_inlines = super().get_inline_instances(request, obj)
  370. inlines = []
  371. if obj and obj.category == IPResource.CATEGORY_PUBLIC:
  372. if obj.allocations.filter(get_active_filter()).exists():
  373. inlines += [ActiveServiceAllocationInline]
  374. else:
  375. inlines += [NewServiceAllocationInline]
  376. if obj.allocations.exclude(get_active_filter()).exists():
  377. inlines += [InactiveServiceAllocationInline]
  378. return [inline(self.model, self.admin_site) for inline in inlines] + super_inlines
  379. def get_queryset(self, request):
  380. qs = super().get_queryset(request)
  381. now = timezone.now()
  382. qs = qs.annotate(
  383. last_use=models.Case(
  384. models.When(in_use=True, then=now),
  385. models.When(category=0, then=models.Max('service_allocation__end')),
  386. models.When(category=1, then=models.Max('antenna_allocation__end')),
  387. default=None,
  388. ))
  389. qs = qs.annotate(
  390. downtime=models.Case(
  391. models.When(last_state__state=IPResourceState.STATE_UP, then=models.F('last_state__date')-models.Value(now)),
  392. models.When(last_state__state=IPResourceState.STATE_DOWN, then=models.Value(now)-models.F('last_time_up')),
  393. default=None,
  394. output_field=models.DurationField(),
  395. ))
  396. qs = qs.annotate(
  397. route=models.Case(
  398. models.When(
  399. in_use_by_service=True,
  400. then=models.Subquery(
  401. ServiceAllocation.objects.filter(
  402. Q(resource=models.OuterRef('pk')) & get_active_filter()
  403. ).values('route__name')[:1]
  404. ),
  405. ),
  406. output_field=models.CharField(),
  407. ))
  408. return qs
  409. def available_display(self, obj):
  410. return not obj.reserved and not obj.in_use
  411. available_display.short_description = 'Disponible'
  412. available_display.boolean = True
  413. def last_use(self, obj):
  414. if obj.last_use:
  415. return naturaltime(obj.last_use)
  416. else:
  417. return '-'
  418. last_use.short_description = 'Dernière utilisation'
  419. last_use.admin_order_field = 'last_use'
  420. def ping(self, obj):
  421. if obj.last_state.state == IPResourceState.STATE_UP:
  422. label = 'UP'
  423. else:
  424. if obj.last_time_up:
  425. label = 'dernier ping : ' + naturaltime(obj.last_time_up)
  426. else:
  427. label = 'DOWN'
  428. if obj.checkmk_url:
  429. return format_html('<a href="{}">{}</a>', obj.checkmk_url, label)
  430. else:
  431. return label
  432. ping.short_description = 'ping'
  433. #ping.admin_order_field = 'last_state__date'
  434. ping.admin_order_field = 'downtime'
  435. def route(self, obj):
  436. return obj.route
  437. route.short_description = 'route'
  438. route.admin_order_field = 'route'
  439. def checkmk(self, obj):
  440. return format_html('<a href="{}">{}</a>', obj.checkmk_url, 'voir')
  441. checkmk.short_description = 'CheckMK'
  442. def contact_ip_owners(self, request, queryset):
  443. selected = request.POST.getlist(admin.ACTION_CHECKBOX_NAME)
  444. services = ServiceAllocation.objects.filter(resource__ip__in=selected) \
  445. .filter(get_active_filter()) \
  446. .values_list('service__adhesion', flat=True)
  447. antennas = AntennaAllocation.objects.filter(resource__ip__in=selected) \
  448. .filter(get_active_filter()) \
  449. .values_list('antenna__contact', flat=True)
  450. pk = ",".join(map(str, set(services) | set(antennas)))
  451. return HttpResponseRedirect(reverse('admin:contact-adherents') + "?pk=%s" % pk)
  452. contact_ip_owners.short_description = 'Contacter les adhérents'
  453. def stop_allocation(self, request, resource):
  454. resource = self.get_object(request, resource)
  455. allocation = resource.allocations.filter(get_active_filter()).first()
  456. if not allocation: # L’IP n’est pas allouée
  457. return HttpResponseRedirect(reverse('admin:services_ipresource_change', args=[resource.pk]))
  458. form = StopAllocationForm(request.POST or None)
  459. if request.method == 'POST' and form.is_valid():
  460. self.message_user(request, 'Allocation stoppée.')
  461. allocation.end = timezone.now()
  462. allocation.save()
  463. notify_allocation(request, allocation)
  464. # Il faudrait rajouter un redirect dans l’URL pour rediriger vers l’IP ou le Service
  465. return HttpResponseRedirect(reverse('admin:services_ipresource_change', args=[resource.pk]))
  466. context = self.admin_site.each_context(request)
  467. context.update({
  468. 'opts': self.model._meta,
  469. 'title': 'Stopper une allocation',
  470. 'object': resource,
  471. 'media': self.media,
  472. 'form': form,
  473. })
  474. return TemplateResponse(request, "admin/services/ipresource/stop_allocation.html", context)
  475. def get_urls(self):
  476. my_urls = [
  477. path('<resource>/stop/', self.admin_site.admin_view(self.stop_allocation), name='stop-allocation'),
  478. ]
  479. return my_urls + super().get_urls()
  480. def get_actions(self, request):
  481. actions = super().get_actions(request)
  482. if 'delete_selected' in actions:
  483. del actions['delete_selected']
  484. return actions
  485. def has_add_permission(self, request, obj=None):
  486. return False
  487. def has_delete_permission(self, request, obj=None):
  488. return False
  489. class IPResourceStateAdmin(admin.ModelAdmin):
  490. list_display = ('ip', 'date', 'state',)
  491. def get_actions(self, request):
  492. actions = super().get_actions(request)
  493. if 'delete_selected' in actions:
  494. del actions['delete_selected']
  495. return actions
  496. def has_add_permission(self, request, obj=None):
  497. return False
  498. def has_change_permission(self, request, obj=None):
  499. return False
  500. def has_delete_permission(self, request, obj=None):
  501. return False
  502. class RouteAdmin(admin.ModelAdmin):
  503. list_display = ('name',)
  504. search_fields = ('name',)
  505. def get_fieldsets(self, request, obj=None):
  506. if obj:
  507. return (
  508. (None, {'fields': ['name']}),
  509. ('Adhérent·e·s', {'fields': ['get_adh'], 'classes': ['collapse']}),
  510. ('E-mails', {'fields': ['get_email'], 'classes': ['collapse']}),
  511. ('SMS', {'fields': ['get_sms'], 'classes': ['collapse']}),
  512. ('IP', {'fields': ['get_ip'], 'classes': ['collapse']}),
  513. )
  514. else:
  515. return (
  516. (None, {'fields': ['name']}),
  517. )
  518. def get_readonly_fields(self, request, obj=None):
  519. if obj:
  520. return ('get_email', 'get_sms', 'get_ip', 'get_adh',)
  521. else:
  522. return ()
  523. def get_email(self, route):
  524. return '\n'.join(route.get_email())
  525. get_email.short_description = 'E-mails'
  526. def get_sms(self, route):
  527. sms_filter = lambda x: x[:2] == '06' or x[:2] == '07' or x[:3] == '+336' or x[:3] == '+337'
  528. return '\n'.join(filter(sms_filter, route.get_tel()))
  529. get_sms.short_description = 'SMS'
  530. def get_ip(self, route):
  531. return '\n'.join(route.get_ip())
  532. get_ip.short_description = 'IP'
  533. def get_adh(self, route):
  534. return '\n'.join(map(lambda adh: '%s %s' % (adh, adh.adherent), route.get_adh()))
  535. get_adh.short_description = 'Adhérent·e·s'
  536. def get_actions(self, request):
  537. actions = super().get_actions(request)
  538. if 'delete_selected' in actions:
  539. del actions['delete_selected']
  540. return actions
  541. def has_delete_permission(self, request, obj=None):
  542. if obj:
  543. if obj.allocations.exists():
  544. return False
  545. return True
  546. return False
  547. class TunnelAdmin(admin.ModelAdmin):
  548. list_display = ('name', 'description', 'created', 'active')
  549. list_filter = (
  550. ActiveTunnelFilter,
  551. )
  552. def get_actions(self, request):
  553. actions = super().get_actions(request)
  554. if 'delete_selected' in actions:
  555. del actions['delete_selected']
  556. return actions
  557. def has_delete_permission(self, request, obj=None):
  558. if obj:
  559. if obj.allocations.exists():
  560. return False
  561. return True
  562. return False
  563. def active(self, obj):
  564. return not obj.ended
  565. active.short_description = 'Actif'
  566. active.boolean = True
  567. class ServiceTypeAdmin(admin.ModelAdmin):
  568. fields = ('name', 'contact')
  569. def get_actions(self, request):
  570. actions = super().get_actions(request)
  571. if 'delete_selected' in actions:
  572. del actions['delete_selected']
  573. return actions
  574. def has_delete_permission(self, request, obj=None):
  575. return False
  576. class ActiveAntennaLayer(GeoJSONLayerView):
  577. def get_queryset(self):
  578. return Antenna.objects.filter(get_active_filter('allocation'))
  579. class AntennaAdmin(admin.ModelAdmin):
  580. #inlines = (ActiveAntennaAllocationInline, InactiveAntennaAllocationInline,)
  581. list_filter = (
  582. AntennaPrefixFilter,
  583. AntennaPositionFilter,
  584. 'mode',
  585. 'ssid',
  586. )
  587. list_display_links = ('id', 'label')
  588. search_fields = ('=id', 'label', 'notes', 'ssid')
  589. raw_id_fields = ('contact',)
  590. form = AntennaForm
  591. def get_queryset(self, request):
  592. qs = super().get_queryset(request)
  593. if connection.vendor == 'postgresql':
  594. qs = qs.annotate(
  595. ip=StringAgg( # concaténation des IP avec des virgules directement par postgresql
  596. Cast( # casting en TextField car StringApp oppère sur des string mais les ip sont des inet
  597. models.Case( # seulement les IP des allocations actives
  598. models.When(
  599. get_active_filter('allocation'),
  600. then='allocation__resource__ip'
  601. ),
  602. ),
  603. models.TextField()
  604. ),
  605. delimiter=', '
  606. )
  607. )
  608. return qs
  609. def get_list_display(self, request):
  610. # ssid_display needs request to access query string and preserve filters
  611. ssid_display = partial(self.ssid_display, request=request)
  612. update_wrapper(ssid_display, self.ssid_display)
  613. return ('id', 'label', 'mode', ssid_display, 'position_display', 'ip_display')
  614. def position_display(self, obj):
  615. return obj.position is not None
  616. position_display.short_description = 'Géolocalisé'
  617. position_display.boolean = True
  618. def ip_display(self, obj):
  619. if connection.vendor == 'postgresql':
  620. return obj.ip
  621. else:
  622. # peu efficace car génère une requête par ligne
  623. allocations = obj.allocations.filter(active=True)
  624. return ', '.join(allocations.values_list('resource__ip', flat=True)) or '-'
  625. ip_display.short_description = 'IP'
  626. def ssid_display(self, obj, request):
  627. if obj.ssid:
  628. qs = request.GET.copy()
  629. qs.update({'ssid': obj.ssid})
  630. ssid_url = reverse('admin:services_antenna_changelist') + '?' + urlencode(qs)
  631. return format_html(u'<a href="{}">{}</a>', ssid_url, obj.ssid)
  632. else:
  633. return None
  634. ssid_display.short_description = 'SSID'
  635. def get_actions(self, request):
  636. actions = super().get_actions(request)
  637. if 'delete_selected' in actions:
  638. del actions['delete_selected']
  639. return actions
  640. def has_delete_permission(self, request, obj=None):
  641. return False
  642. def get_urls(self):
  643. my_urls = [
  644. url(r'^map/$', self.admin_site.admin_view(self.map_view, cacheable=True), name='antenna-map'),
  645. url(r'^map/data.json$', self.admin_site.admin_view(ActiveAntennaLayer.as_view(
  646. model=Antenna,
  647. geometry_field='position',
  648. properties=('label', 'mode', 'ssid', 'orientation', 'absolute_url',),
  649. )), name='antenna-map-data'),
  650. ]
  651. return my_urls + super().get_urls()
  652. def map_view(self, request):
  653. return TemplateResponse(request, 'services/antenna_map.html', dict(
  654. self.admin_site.each_context(request),
  655. opts=self.model._meta,
  656. json_url=reverse('admin:antenna-map-data'),
  657. ))
  658. def map_data_view(self, request):
  659. geojson = serialize('geojson', Antenna.objects.all(), geometry_field='point', fields=('position',))
  660. return HttpResponse(geojson, content_type='application/json')
  661. class SwitchAdmin(admin.ModelAdmin):
  662. list_display = ('name', 'ports_count', 'active_ports_count', 'inactive_ports_count', 'unknown_ports_count',)
  663. fields = ('name', 'first_port', 'last_port', 'notes',)
  664. search_fields = ('name', 'notes', 'ports__notes', 'ports__service__label',)
  665. def get_queryset(self, request):
  666. qs = super().get_queryset(request)
  667. qs = qs.annotate(
  668. active_ports_count=models.Count(models.Case(models.When(ports__up=True, then=models.Value('1')))),
  669. unknown_ports_count=models.Count(models.Case(models.When(ports__up__isnull=True, then=models.Value('1')))),
  670. inactive_ports_count=models.Count(models.Case(models.When(ports__up=False, then=models.Value('1')))),
  671. )
  672. return qs
  673. def ports_count(self, switch):
  674. return switch.last_port - switch.first_port + 1
  675. ports_count.short_description = 'Nombre de ports'
  676. def active_ports_count(self, switch):
  677. return switch.active_ports_count
  678. active_ports_count.short_description = 'up'
  679. active_ports_count.admin_order_field = 'active_ports_count'
  680. def inactive_ports_count(self, switch):
  681. return switch.inactive_ports_count
  682. inactive_ports_count.short_description = 'down'
  683. inactive_ports_count.admin_order_field = 'inactive_ports_count'
  684. def unknown_ports_count(self, switch):
  685. return switch.unknown_ports_count
  686. unknown_ports_count.short_description = 'inconnus'
  687. unknown_ports_count.admin_order_field = 'unknown_ports_count'
  688. def get_inline_instances(self, request, obj=None):
  689. if obj:
  690. return [ SwitchPortInline(self.model, self.admin_site) ]
  691. else:
  692. return []
  693. def get_readonly_fields(self, request, obj=None):
  694. if obj:
  695. return ('first_port', 'last_port',)
  696. else:
  697. return ()
  698. def get_actions(self, request):
  699. actions = super().get_actions(request)
  700. if 'delete_selected' in actions:
  701. del actions['delete_selected']
  702. return actions
  703. def has_delete_permission(self, request, obj=None):
  704. return False
  705. admin.site.register(ServiceType, ServiceTypeAdmin)
  706. admin.site.register(Service, ServiceAdmin)
  707. admin.site.register(IPPrefix, IPPrefixAdmin)
  708. admin.site.register(IPResource, IPResourceAdmin)
  709. admin.site.register(Route, RouteAdmin)
  710. admin.site.register(Tunnel, TunnelAdmin)
  711. admin.site.register(Antenna, AntennaAdmin)
  712. admin.site.register(Switch, SwitchAdmin)