from django.db import models

from django.contrib.auth.models import User 

from django.contrib.auth.models import Group, Permission

PROBLEM_CHOICES=(

    ('P', 'Plotting'),

    ('S', 'Scripting'),

)

SESSION_CHOICES=(

  ('A','DAY 1 SESSION 1'),

  ('B','DAY 1 SESSION 2'),

  ('C','DAY 1 SESSION 3'),

  ('D','DAY 1 SESSION 4'),

  ('E','DAY 2 SESSION 1'),

  ('F','DAY 2 SESSION 2'),

  ('G','DAY 2 SESSION 3'),

  ('H','DAY 1 SESSION 4'),

  )

class Problem(models.Model):

  # name = models.CharField(blank=False,max_length=255)

  description = models.CharField(blank=False , max_length=255)

  problem_type=models.CharField(blank = False ,max_length=1, choices=PROBLEM_CHOICES)

  solution = models.CharField(blank=True ,max_length=255)

  solution_image = models.ImageField(blank = True ,upload_to='plots')

  session = models.CharField(blank=False ,max_length=1,choices=SESSION_CHOICES)

  author = models.CharField(max_length=255)

  created = models.DateTimeField()

  modified = models.DateTimeField()

  credit=models.IntegerField()

  #if  user.has_perm('testapp.add_bar') and  user.has_perm('foo.change_bar') and user.has_perm('foo.delete_bar')

class Test_User(models.Model):

  #problem = models.ForeignKey('Problem')

  user = models.ForeignKey(User, unique=True)

class Score(models.Model):

  user = models.ForeignKey(User, unique=True)

  total_credits=models.IntegerField(default=0,blank=False)