← Back to branch summary

~bnrubin/bantrackertwo/devel

« back to all changes in this revision

Viewing changes to bt/models.py

  • Committer: Benjamin Rubin
  • Date: 2009-09-02 02:14:31 UTC
  • Revision ID: bnrubin@romulus-20090902021431-02x226yec1d5sxal
- Cleaned up Event code and presentation
- Added basic Issue and Settings forms and presentation

Show diffs side-by-side

added added

removed removed

Lines of Context:
bt/models.py
3
3
from django.db.models import signals
4
4
from django.contrib.contenttypes.models import ContentType
5
5
from django.contrib.contenttypes import generic
 
6
from pytz import common_timezones
 
7
from django import forms
 
8
from django.forms import ModelForm
6
9
import re
7
10
 
8
 
# Create your models here.
9
 
# 
10
 
# Todo:
11
 
# - Should channels have their own models? Useful if we intend to user per channel permissions
12
 
# - Will need to figure out how to assign events for ops who may not have logins (like freenode staffers)
13
 
# - Determine how kickbans will be logged
14
 
# - Take a look at django's built-in comment model
 
11
# Make a dict of all timezones using pytz instead of maintaining a static dict here
 
12
TZ_CHOICES = []
 
13
for zone in common_timezones:
 
14
    TZ_CHOICES.append((zone,zone))
 
15
 
15
16
 
16
17
class Issue(models.Model):
17
18
    create_date = models.DateTimeField(auto_now_add = True,editable = False)
18
19
    modify_date = models.DateTimeField(auto_now = True, editable = False)
19
20
    closed_date = models.DateTimeField(null = True)
20
21
    author = models.ForeignKey(User,null = True)
 
22
    description = models.TextField()
 
23
    log = models.ForeignKey('Log', null = True)
21
24
    comments = generic.GenericRelation('Comment')
 
25
 
22
26
    def is_closed(self):
23
27
        return self.closed_date != None
24
28
 
 
29
class IssueForm(ModelForm):
 
30
    class Meta:
 
31
        model = Issue
 
32
        fields = ('description',)
 
33
    
 
34
 
 
35
 
25
36
class Event(models.Model):
26
37
    "Defines the Event base class. Every IRC event class should inherit from this."
27
38
    create_date = models.DateTimeField(auto_now_add = False)
30
41
    log = models.ForeignKey('Log')
31
42
    operator = models.ForeignKey('Operator')
32
43
    hostmask = models.CharField(max_length=255)
33
 
    comments = generic.GenericRelation('Comment') 
 
44
    comments = generic.GenericRelation('Comment')
34
45
    
35
46
    def __unicode__(self):
36
47
        return self.hostmask
42
53
class Ban(Event):
43
54
    mask = models.CharField(max_length=255)
44
55
    removal_date = models.DateTimeField(null = True)
45
 
    removed_by = models.ForeignKey('Operator', null = True)
46
 
    
 
56
    removed_by = models.ForeignKey('Operator', null = True, related_name = 'removed_by')
 
57
    kick = models.OneToOneField('Kick', null = True)
47
58
    
48
59
    def is_removed(self):
49
60
        return self.removal_date != None
50
61
 
 
62
    def is_mute(self):
 
63
        return self.mask[0] == '%'
 
64
 
51
65
    def __unicode__(self):
52
66
        return self.mask
53
67
 
56
70
   
57
71
class Kick(Event):
58
72
    reason = models.CharField(max_length=255,null = True) 
 
73
    
59
74
    def __unicode__(self):
60
75
        return self.hostmask
61
76
    
62
77
class Mark(Event):
63
78
    reason = models.CharField(max_length=255)
 
79
    
64
80
    def __unicode__(self):
65
81
        return self.reason
66
82
 
78
94
class UserProfile(models.Model):
79
95
    user = models.ForeignKey(User, unique = True, editable=False)
80
96
    hostmask = models.CharField(max_length=255) # Will need to figure out how to lock this down, so that people cannot assign themselves *!*@*
 
97
    timezone = models.CharField(max_length=100,choices=TZ_CHOICES)
81
98
    
82
99
    def hostmask_pattern(self):
83
100
        return re.compile(self.hostmask.replace('?','.').replace('*','.*'))