from django.db import models
from django.utils.translation import ugettext as _

COUNTRIES = (
    ('BE', _('Belgium')),
    ('BG', _('Bulgaria')),
    ('CZ', _('Czech Republic')),
    ('DK', _('Denmark')),
    ('DE', _('Germany')),
    ('EE', _('Estonia')),
    ('IE', _('Ireland')),
    ('EL', _('Greece')),
    ('ES', _('Spain')),
    ('FR', _('France')),
    ('IT', _('Italy')),
    ('CY', _('Cyprus')),
    ('LV', _('Latvia')),
    ('LT', _('Lithuania')),
    ('LU', _('Luxembourg')),
    ('HU', _('Hungary')),
    ('MT', _('Malta')),
    ('NL', _('Netherlands')),
    ('AT', _('Austria')),
    ('PL', _('Poland')),
    ('PT', _('Portugal')),
    ('RO', _('Romania')),
    ('SI', _('Slovenia')),
    ('SK', _('Slovakia')),
    ('FI', _('Finland')),
    ('SE', _('Sweden')),
    ('UK', _('United Kingdom ')),
    )

RESOURCES = (
    ('port', _('port')),
    ('protocol', _('protocol')),
    ('service', _('service')),
    ('site', _('site')),
    ('user', _('user')),
    ('ip', _('ip')),
    )
TYPES = (
    ('blocking', _('Blocking')),
    ('throttling', _('Throttling')),
    )
MEDIA = (
    ('fixed', _('Fixed')),
    ('mobile', _('Mobile')),
    )

class Violation(models.Model):
    country = models.CharField(max_length=2, choices=COUNTRIES)
    operator = models.CharField(max_length=256)
    contract = models.CharField(max_length=256)
    resource = models.CharField(max_length=20, choices=RESOURCES)
    resource_name = models.CharField(max_length=4096)
    type = models.CharField(max_length=20, choices=TYPES)
    media = models.CharField( max_length=20, choices=MEDIA)
    temporary = models.BooleanField( )
    contractual = models.BooleanField()
    contract_excerpt = models.TextField()
    loophole = models.BooleanField()

class Comment(models.Model):
    submitter_email = models.EmailField()
    submitter_name = models.CharField(max_length=20)
    comment = models.TextField()
    timestamp = models.DateTimeField()
    violation = models.ForeignKey(Violation)

class Attachment(models.Model):
    storage = models.FileField(upload_to='static')
    comment = models.ForeignKey(Comment)