import os
from django.db import models
from ghostnames.utils import tuplify_list
GHOST_NAME_AVAILABILITY = tuplify_list(['taken','available'])
class Ghost(models.Model):
name = models.CharField(max_length=255)
taken = models.CharField(max_length=32, choices=GHOST_NAME_AVAILABILITY)
class Meta:
ordering = ['id']
def __unicode__(self):
return self.name
@classmethod
def initialize(self, filepath= os.path.join(os.path.dirname(os.path.realpath(__file__)),
'initial_data/initial_data.txt')):
if Ghost.objects.count() == 0:
with open(filepath) as f:
for ghostname in f:
Ghost.objects.create(name=ghostname.strip(),
taken='available')
class Username(models.Model):
firstname = models.CharField(max_length=255)
lastname = models.CharField(max_length=255)
ghostname = models.CharField(max_length=255,default='')
