Races: add module to do a crude racebot thing
join/start/update/finish/leave. good enough start i think
This commit is contained in:
parent
3316f4a868
commit
5fff9411a5
|
|
@ -38,6 +38,7 @@ INSTALLED_APPS = (
|
|||
'django.contrib.staticfiles',
|
||||
'django_extensions',
|
||||
'south',
|
||||
'races',
|
||||
'seen',
|
||||
)
|
||||
|
||||
|
|
|
|||
|
|
@ -0,0 +1,209 @@
|
|||
"""
|
||||
Seen - track races and those racing them
|
||||
Copyright (C) 2014 Brian S. Stephan
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
"""
|
||||
|
||||
import re
|
||||
|
||||
from races.models import Race, Racer, RaceUpdate
|
||||
|
||||
from Module import Module
|
||||
|
||||
|
||||
class Races(Module):
|
||||
|
||||
"""Track races and allow for logging updates."""
|
||||
|
||||
def register_handlers(self):
|
||||
"""Hook handler functions into the IRC library."""
|
||||
|
||||
self.irc.add_global_regex_handler(['pubmsg', 'privmsg'],
|
||||
r'^!race\s+(\S+)\s+join$',
|
||||
self.join)
|
||||
self.irc.add_global_regex_handler(['pubmsg', 'privmsg'],
|
||||
r'^!race\s+(\S+)\s+leave$',
|
||||
self.leave)
|
||||
self.irc.add_global_regex_handler(['pubmsg', 'privmsg'],
|
||||
r'^!race\s+(\S+)\s+start$',
|
||||
self.start)
|
||||
self.irc.add_global_regex_handler(['pubmsg', 'privmsg'],
|
||||
r'^!race\s+(\S+)\s+finish$',
|
||||
self.finish)
|
||||
self.irc.add_global_regex_handler(['pubmsg', 'privmsg'],
|
||||
r'^!race\s+(\S+)\s+update\s+(.*)$',
|
||||
self.update)
|
||||
|
||||
def unregister_handlers(self):
|
||||
"""Unhook handler functions into the IRC library."""
|
||||
|
||||
self.irc.remove_global_regex_handler(['pubmsg', 'privmsg'],
|
||||
r'^!race\s+(\S+)\s+join$',
|
||||
self.join)
|
||||
self.irc.remove_global_regex_handler(['pubmsg', 'privmsg'],
|
||||
r'^!race\s+(\S+)\s+leave$',
|
||||
self.leave)
|
||||
self.irc.remove_global_regex_handler(['pubmsg', 'privmsg'],
|
||||
r'^!race\s+(\S+)\s+start$',
|
||||
self.start)
|
||||
self.irc.remove_global_regex_handler(['pubmsg', 'privmsg'],
|
||||
r'^!race\s+(\S+)\s+finish$',
|
||||
self.finish)
|
||||
self.irc.remove_global_regex_handler(['pubmsg', 'privmsg'],
|
||||
r'^!race\s+(\S+)\s+update\s+(.*)$',
|
||||
self.update)
|
||||
|
||||
def join(self, nick, userhost, event, from_admin, groups):
|
||||
"""Join a race."""
|
||||
|
||||
race_id, = groups
|
||||
try:
|
||||
race = Race.objects.get(key=race_id)
|
||||
|
||||
# is the nick already in the race
|
||||
try:
|
||||
racer = Racer.objects.get(race=race, nick=nick)
|
||||
|
||||
# ideally they're not in the db, but if they aren't flagged as joined,
|
||||
# then let them join again i guess
|
||||
if racer.joined:
|
||||
return self.irc.reply(event, "you are already in {0:s}.".format(race.name))
|
||||
except Racer.DoesNotExist:
|
||||
racer = Racer()
|
||||
racer.race = race
|
||||
racer.nick = nick
|
||||
|
||||
racer.joined = True
|
||||
racer.started = False
|
||||
racer.finished = False
|
||||
racer.save()
|
||||
|
||||
return self.irc.reply(event,
|
||||
"you have joined {0:s}.".format(race.name))
|
||||
except Race.DoesNotExist:
|
||||
return self.irc.reply(event, "race {0:s} not found.".format(race_id))
|
||||
|
||||
def leave(self, nick, userhost, event, from_admin, groups):
|
||||
"""Leave a race."""
|
||||
|
||||
race_id, = groups
|
||||
try:
|
||||
race = Race.objects.get(key=race_id)
|
||||
|
||||
try:
|
||||
racer = Racer.objects.get(race=race, nick=nick)
|
||||
|
||||
if racer.finished:
|
||||
return self.irc.reply(event, "you can't leave {0:s} since you finished it."
|
||||
"".format(race.name))
|
||||
|
||||
if racer.joined:
|
||||
racer.joined = False
|
||||
racer.save()
|
||||
|
||||
return self.irc.reply(event, "you have left {0:s}.".format(race.name))
|
||||
else:
|
||||
return self.irc.reply(event, "you weren't in {0:s}.".format(race.name))
|
||||
except Racer.DoesNotExist:
|
||||
return self.irc.reply(event, "you weren't in {0:s}.".format(race.name))
|
||||
except Race.DoesNotExist:
|
||||
return self.irc.reply(event, "race {0:s} not found.".format(race_id))
|
||||
|
||||
def start(self, nick, userhost, event, from_admin, groups):
|
||||
"""Start racing."""
|
||||
|
||||
race_id, = groups
|
||||
try:
|
||||
race = Race.objects.get(key=race_id)
|
||||
|
||||
try:
|
||||
racer = Racer.objects.get(race=race, nick=nick)
|
||||
|
||||
if racer.started:
|
||||
return self.irc.reply(event, "you have already started {0:s}.".format(race.name))
|
||||
|
||||
if racer.finished:
|
||||
return self.irc.reply(event, "you can't start {0:s} since you finished it."
|
||||
"".format(race.name))
|
||||
|
||||
racer.started = True
|
||||
racer.save()
|
||||
|
||||
# create an event for this too
|
||||
RaceUpdate.objects.create(racer=racer, update="Joined the race.")
|
||||
|
||||
return self.irc.reply(event, "you have started {0:s}.".format(race.name))
|
||||
except Racer.DoesNotExist:
|
||||
return self.irc.reply(event, "you have not joined {0:s}.".format(race.name))
|
||||
except Race.DoesNotExist:
|
||||
return self.irc.reply(event, "race {0:s} not found.".format(race_id))
|
||||
|
||||
def finish(self, nick, userhost, event, from_admin, groups):
|
||||
"""Finish racing."""
|
||||
|
||||
race_id, = groups
|
||||
try:
|
||||
race = Race.objects.get(key=race_id)
|
||||
|
||||
try:
|
||||
racer = Racer.objects.get(race=race, nick=nick)
|
||||
|
||||
if not racer.started:
|
||||
return self.irc.reply(event, "you have not started {0:s}.".format(race.name))
|
||||
|
||||
if racer.finished:
|
||||
return self.irc.reply(event, "you have already finished {0:s}."
|
||||
"".format(race.name))
|
||||
|
||||
racer.finished = True
|
||||
racer.save()
|
||||
|
||||
# create an event for this too
|
||||
RaceUpdate.objects.create(racer=racer, update="Finished the race.")
|
||||
|
||||
return self.irc.reply(event, "you have finished {0:s}.".format(race.name))
|
||||
except Racer.DoesNotExist:
|
||||
return self.irc.reply(event, "you have not joined {0:s}.".format(race.name))
|
||||
except Race.DoesNotExist:
|
||||
return self.irc.reply(event, "race {0:s} not found.".format(race_id))
|
||||
|
||||
def update(self, nick, userhost, event, from_admin, groups):
|
||||
"""Add an update to an ongoing race."""
|
||||
|
||||
race_id, update = groups
|
||||
try:
|
||||
race = Race.objects.get(key=race_id)
|
||||
|
||||
try:
|
||||
racer = Racer.objects.get(race=race, nick=nick)
|
||||
|
||||
if not racer.started:
|
||||
return self.irc.reply(event, "you have not started {0:s}.".format(race.name))
|
||||
|
||||
if racer.finished:
|
||||
return self.irc.reply(event, "you have already finished {0:s}."
|
||||
"".format(race.name))
|
||||
|
||||
# create an event
|
||||
RaceUpdate.objects.create(racer=racer, update=update)
|
||||
|
||||
return self.irc.reply(event, "update logged.")
|
||||
except Racer.DoesNotExist:
|
||||
return self.irc.reply(event, "you have not joined {0:s}.".format(race.name))
|
||||
except Race.DoesNotExist:
|
||||
return self.irc.reply(event, "race {0:s} not found.".format(race_id))
|
||||
|
||||
# vi:tabstop=4:expandtab:autoindent
|
||||
|
|
@ -0,0 +1,9 @@
|
|||
from django.contrib import admin
|
||||
|
||||
from races.models import Race, Racer, RaceUpdate
|
||||
|
||||
admin.site.register(Race)
|
||||
admin.site.register(Racer)
|
||||
admin.site.register(RaceUpdate)
|
||||
|
||||
# vi:tabstop=4:expandtab:autoindent
|
||||
|
|
@ -0,0 +1,82 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
from south.utils import datetime_utils as datetime
|
||||
from south.db import db
|
||||
from south.v2 import SchemaMigration
|
||||
from django.db import models
|
||||
|
||||
|
||||
class Migration(SchemaMigration):
|
||||
|
||||
def forwards(self, orm):
|
||||
# Adding model 'Race'
|
||||
db.create_table(u'races_race', (
|
||||
('key', self.gf('django.db.models.fields.CharField')(max_length=16, primary_key=True)),
|
||||
('name', self.gf('django.db.models.fields.CharField')(max_length=255)),
|
||||
('description', self.gf('django.db.models.fields.TextField')()),
|
||||
))
|
||||
db.send_create_signal(u'races', ['Race'])
|
||||
|
||||
# Adding model 'Racer'
|
||||
db.create_table(u'races_racer', (
|
||||
(u'id', self.gf('django.db.models.fields.AutoField')(primary_key=True)),
|
||||
('nick', self.gf('django.db.models.fields.CharField')(max_length=64)),
|
||||
('race', self.gf('django.db.models.fields.related.ForeignKey')(to=orm['races.Race'])),
|
||||
('joined', self.gf('django.db.models.fields.BooleanField')(default=False)),
|
||||
('started', self.gf('django.db.models.fields.BooleanField')(default=False)),
|
||||
('finished', self.gf('django.db.models.fields.BooleanField')(default=False)),
|
||||
))
|
||||
db.send_create_signal(u'races', ['Racer'])
|
||||
|
||||
# Adding unique constraint on 'Racer', fields ['nick', 'race']
|
||||
db.create_unique(u'races_racer', ['nick', 'race_id'])
|
||||
|
||||
# Adding model 'RaceUpdate'
|
||||
db.create_table(u'races_raceupdate', (
|
||||
(u'id', self.gf('django.db.models.fields.AutoField')(primary_key=True)),
|
||||
('racer', self.gf('django.db.models.fields.related.ForeignKey')(to=orm['races.Racer'])),
|
||||
('update', self.gf('django.db.models.fields.TextField')()),
|
||||
('event_time', self.gf('django.db.models.fields.DateTimeField')(default=datetime.datetime.now)),
|
||||
))
|
||||
db.send_create_signal(u'races', ['RaceUpdate'])
|
||||
|
||||
|
||||
def backwards(self, orm):
|
||||
# Removing unique constraint on 'Racer', fields ['nick', 'race']
|
||||
db.delete_unique(u'races_racer', ['nick', 'race_id'])
|
||||
|
||||
# Deleting model 'Race'
|
||||
db.delete_table(u'races_race')
|
||||
|
||||
# Deleting model 'Racer'
|
||||
db.delete_table(u'races_racer')
|
||||
|
||||
# Deleting model 'RaceUpdate'
|
||||
db.delete_table(u'races_raceupdate')
|
||||
|
||||
|
||||
models = {
|
||||
u'races.race': {
|
||||
'Meta': {'object_name': 'Race'},
|
||||
'description': ('django.db.models.fields.TextField', [], {}),
|
||||
'key': ('django.db.models.fields.CharField', [], {'max_length': '16', 'primary_key': 'True'}),
|
||||
'name': ('django.db.models.fields.CharField', [], {'max_length': '255'})
|
||||
},
|
||||
u'races.racer': {
|
||||
'Meta': {'unique_together': "(('nick', 'race'),)", 'object_name': 'Racer'},
|
||||
'finished': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
|
||||
u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
|
||||
'joined': ('django.db.models.fields.BooleanField', [], {'default': 'False'}),
|
||||
'nick': ('django.db.models.fields.CharField', [], {'max_length': '64'}),
|
||||
'race': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['races.Race']"}),
|
||||
'started': ('django.db.models.fields.BooleanField', [], {'default': 'False'})
|
||||
},
|
||||
u'races.raceupdate': {
|
||||
'Meta': {'object_name': 'RaceUpdate'},
|
||||
'event_time': ('django.db.models.fields.DateTimeField', [], {'default': 'datetime.datetime.now'}),
|
||||
u'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}),
|
||||
'racer': ('django.db.models.fields.related.ForeignKey', [], {'to': u"orm['races.Racer']"}),
|
||||
'update': ('django.db.models.fields.TextField', [], {})
|
||||
}
|
||||
}
|
||||
|
||||
complete_apps = ['races']
|
||||
|
|
@ -0,0 +1,54 @@
|
|||
from django.db import models
|
||||
from django.utils import timezone
|
||||
|
||||
|
||||
class Race(models.Model):
|
||||
|
||||
"""Track a race."""
|
||||
|
||||
key = models.CharField(max_length=16, primary_key=True)
|
||||
name = models.CharField(max_length=255)
|
||||
description = models.TextField()
|
||||
|
||||
def __unicode__(self):
|
||||
"""Text representation of a race."""
|
||||
|
||||
return u"{0:s} ({1:s})".format(self.name, self.key)
|
||||
|
||||
|
||||
class Racer(models.Model):
|
||||
|
||||
"""Track a racer in a race."""
|
||||
|
||||
nick = models.CharField(max_length=64)
|
||||
race = models.ForeignKey(Race)
|
||||
|
||||
joined = models.BooleanField(default=False)
|
||||
started = models.BooleanField(default=False)
|
||||
finished = models.BooleanField(default=False)
|
||||
|
||||
class Meta:
|
||||
unique_together = ('nick', 'race')
|
||||
|
||||
def __unicode__(self):
|
||||
"""Text representation of a race racer."""
|
||||
|
||||
return u"{0:s} in {1:s}".format(self.nick, self.race.name)
|
||||
|
||||
|
||||
class RaceUpdate(models.Model):
|
||||
|
||||
"""Periodic updates for a racer."""
|
||||
|
||||
racer = models.ForeignKey(Racer)
|
||||
update = models.TextField()
|
||||
event_time = models.DateTimeField(default=timezone.now)
|
||||
|
||||
def __unicode__(self):
|
||||
"""Text representation of a race update."""
|
||||
|
||||
local_time = timezone.localtime(self.event_time)
|
||||
return u"{0:s} in {1:s} @ {2:s}".format(self.racer.nick, self.racer.race.name,
|
||||
local_time.strftime('%Y-%m-%d %H:%M:%S %Z'))
|
||||
|
||||
# vi:tabstop=4:expandtab:autoindent
|
||||
Loading…
Reference in New Issue