from django.contrib import admin
from .models import User, Scan, Scanner, SeverityCount, Asset, Vulnerability

# Register your models here.

class ScanInline(admin.StackedInline):
    model = SeverityCount


class ScanAdmin(admin.ModelAdmin):
    inlines = [ScanInline]


admin.site.register(User)
admin.site.register(Scanner)
admin.site.register(Asset)
admin.site.register(Scan, ScanAdmin)
admin.site.register(Vulnerability)
