mirror of
https://github.com/ae-utbm/sith.git
synced 2024-11-18 12:13:24 +00:00
270 lines
10 KiB
Python
270 lines
10 KiB
Python
from django.db import models, DataError
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.utils import timezone
|
|
from django.conf import settings
|
|
from django.core.urlresolvers import reverse
|
|
from django.forms import ValidationError
|
|
|
|
from datetime import timedelta
|
|
import random
|
|
import string
|
|
|
|
from club.models import Club
|
|
from accounting.models import CurrencyField
|
|
from core.models import Group, User
|
|
from subscription.models import Subscriber
|
|
from subscription.views import get_subscriber
|
|
|
|
class Customer(models.Model):
|
|
"""
|
|
This class extends a user to make a customer. It adds some basic customers informations, such as the accound ID, and
|
|
is used by other accounting classes as reference to the customer, rather than using User
|
|
"""
|
|
user = models.OneToOneField(User, primary_key=True)
|
|
account_id = models.CharField(_('account id'), max_length=10, unique=True)
|
|
amount = CurrencyField(_('amount'))
|
|
|
|
class Meta:
|
|
verbose_name = _('customer')
|
|
verbose_name_plural = _('customers')
|
|
ordering = ['account_id',]
|
|
|
|
def __str__(self):
|
|
return self.user.username
|
|
|
|
def generate_account_id(number):
|
|
number = str(number)
|
|
letter = random.choice(string.ascii_lowercase)
|
|
while Customer.objects.filter(account_id=number+letter).exists():
|
|
letter = random.choice(string.ascii_lowercase)
|
|
return number+letter
|
|
|
|
def save(self, *args, **kwargs):
|
|
if self.amount < 0:
|
|
raise ValidationError(_("Not enough money"))
|
|
super(Customer, self).save(*args, **kwargs)
|
|
|
|
class ProductType(models.Model):
|
|
"""
|
|
This describes a product type
|
|
Useful only for categorizing, changes are made at the product level for now
|
|
"""
|
|
name = models.CharField(_('name'), max_length=30)
|
|
description = models.TextField(_('description'), null=True, blank=True)
|
|
icon = models.ImageField(upload_to='products', null=True, blank=True)
|
|
|
|
class Meta:
|
|
verbose_name = _('product type')
|
|
|
|
def is_owned_by(self, user):
|
|
"""
|
|
Method to see if that object can be edited by the given user
|
|
"""
|
|
if user.is_in_group(settings.SITH_GROUPS['accounting-admin']['name']):
|
|
return True
|
|
return False
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('counter:producttype_list')
|
|
|
|
class Product(models.Model):
|
|
"""
|
|
This describes a product, with all its related informations
|
|
"""
|
|
name = models.CharField(_('name'), max_length=30)
|
|
description = models.TextField(_('description'), blank=True)
|
|
product_type = models.ForeignKey(ProductType, related_name='products', null=True, blank=True)
|
|
code = models.CharField(_('code'), max_length=10)
|
|
purchase_price = CurrencyField(_('purchase price'))
|
|
selling_price = CurrencyField(_('selling price'))
|
|
special_selling_price = CurrencyField(_('special selling price'))
|
|
icon = models.ImageField(upload_to='products', null=True, blank=True)
|
|
club = models.ForeignKey(Club, related_name="products")
|
|
|
|
class Meta:
|
|
verbose_name = _('product')
|
|
|
|
def is_owned_by(self, user):
|
|
"""
|
|
Method to see if that object can be edited by the given user
|
|
"""
|
|
if user.is_in_group(settings.SITH_GROUPS['accounting-admin']['name']):
|
|
return True
|
|
return False
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('counter:product_list')
|
|
|
|
class Counter(models.Model):
|
|
name = models.CharField(_('name'), max_length=30)
|
|
club = models.ForeignKey(Club, related_name="counters")
|
|
products = models.ManyToManyField(Product, related_name="counters", blank=True)
|
|
type = models.CharField(_('subscription type'),
|
|
max_length=255,
|
|
choices=[('BAR',_('Bar')), ('OFFICE',_('Office')), ('EBOUTIC',_('Eboutic'))])
|
|
sellers = models.ManyToManyField(Subscriber, verbose_name=_('sellers'), related_name='counters', blank=True)
|
|
edit_groups = models.ManyToManyField(Group, related_name="editable_counters", blank=True)
|
|
view_groups = models.ManyToManyField(Group, related_name="viewable_counters", blank=True)
|
|
barmen_session = {}
|
|
|
|
class Meta:
|
|
verbose_name = _('counter')
|
|
|
|
def __getattribute__(self, name):
|
|
if name == "edit_groups":
|
|
return Group.objects.filter(name=self.club.unix_name+settings.SITH_BOARD_SUFFIX).all()
|
|
return object.__getattribute__(self, name)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def get_absolute_url(self):
|
|
if self.type == "EBOUTIC":
|
|
return reverse('eboutic:main')
|
|
return reverse('counter:details', kwargs={'counter_id': self.id})
|
|
|
|
def is_owned_by(self, user):
|
|
return user.is_in_group(settings.SITH_GROUPS['counter-admin']['name'])
|
|
|
|
def can_be_viewed_by(self, user):
|
|
if self.type == "BAR" or self.type == "EBOUTIC":
|
|
return True
|
|
sub = get_subscriber(request.user)
|
|
return user.is_in_group(settings.SITH_MAIN_BOARD_GROUP) or sub in self.sellers
|
|
|
|
def add_barman(counter_id, user_id):
|
|
"""
|
|
Logs a barman in to the given counter
|
|
A user is stored as a tuple with its login time
|
|
"""
|
|
counter_id = int(counter_id)
|
|
user_id = int(user_id)
|
|
if counter_id not in Counter.barmen_session.keys():
|
|
Counter.barmen_session[counter_id] = {'users': {(user_id, timezone.now())}, 'time': timezone.now()}
|
|
else:
|
|
Counter.barmen_session[counter_id]['users'].add((user_id, timezone.now()))
|
|
|
|
def del_barman(counter_id, user_id):
|
|
"""
|
|
Logs a barman out and store its permanency
|
|
"""
|
|
counter_id = int(counter_id)
|
|
user_id = int(user_id)
|
|
user_tuple = None
|
|
for t in Counter.barmen_session[counter_id]['users']:
|
|
if t[0] == user_id: user_tuple = t
|
|
Counter.barmen_session[counter_id]['users'].remove(user_tuple)
|
|
u = User.objects.filter(id=user_id).first()
|
|
c = Counter.objects.filter(id=counter_id).first()
|
|
Permanency(user=u, counter=c, start=user_tuple[1], end=Counter.barmen_session[counter_id]['time']).save()
|
|
|
|
def get_barmen_list(counter_id):
|
|
"""
|
|
Returns the barman list as list of User
|
|
|
|
Also handle the timeout of the barmen
|
|
"""
|
|
bl = []
|
|
counter_id = int(counter_id)
|
|
if counter_id in list(Counter.barmen_session.keys()):
|
|
for b in Counter.barmen_session[counter_id]['users']:
|
|
# Reminder: user is stored as a tuple with its login time
|
|
bl.append(User.objects.filter(id=b[0]).first())
|
|
if (timezone.now() - Counter.barmen_session[counter_id]['time']) < timedelta(minutes=settings.SITH_BARMAN_TIMEOUT):
|
|
Counter.barmen_session[counter_id]['time'] = timezone.now()
|
|
else:
|
|
for b in bl:
|
|
Counter.del_barman(counter_id, b.id)
|
|
bl = []
|
|
Counter.barmen_session[counter_id]['users'] = set()
|
|
return bl
|
|
|
|
def get_random_barman(counter_id):
|
|
bl = Counter.get_barmen_list(counter_id)
|
|
return bl[randrange(0, len(bl))]
|
|
|
|
class Refilling(models.Model):
|
|
"""
|
|
Handle the refilling
|
|
"""
|
|
counter = models.ForeignKey(Counter, related_name="refillings", blank=False)
|
|
amount = CurrencyField(_('amount'))
|
|
operator = models.ForeignKey(User, related_name="refillings_as_operator", blank=False)
|
|
customer = models.ForeignKey(Customer, related_name="refillings", blank=False)
|
|
date = models.DateTimeField(_('date'), auto_now=True)
|
|
payment_method = models.CharField(_('payment method'), max_length=255,
|
|
choices=settings.SITH_COUNTER_PAYMENT_METHOD, default='cash')
|
|
bank = models.CharField(_('bank'), max_length=255,
|
|
choices=settings.SITH_COUNTER_BANK, default='other')
|
|
is_validated = models.BooleanField(_('is validated'), default=False)
|
|
|
|
class Meta:
|
|
verbose_name = _("refilling")
|
|
|
|
def __str__(self):
|
|
return "Refilling: %.2f for %s" % (self.amount, self.customer.user.get_display_name())
|
|
|
|
# def get_absolute_url(self):
|
|
# return reverse('counter:details', kwargs={'counter_id': self.id})
|
|
|
|
def save(self, *args, **kwargs):
|
|
self.full_clean()
|
|
if not self.is_validated:
|
|
self.customer.amount += self.amount
|
|
self.customer.save()
|
|
self.is_validated = True
|
|
super(Refilling, self).save(*args, **kwargs)
|
|
|
|
class Selling(models.Model):
|
|
"""
|
|
Handle the sellings
|
|
"""
|
|
label = models.CharField(_("label"), max_length=30)
|
|
product = models.ForeignKey(Product, related_name="sellings", null=True, blank=True)
|
|
counter = models.ForeignKey(Counter, related_name="sellings", blank=False)
|
|
unit_price = CurrencyField(_('unit price'))
|
|
quantity = models.IntegerField(_('quantity'))
|
|
seller = models.ForeignKey(User, related_name="sellings_as_operator", blank=False)
|
|
customer = models.ForeignKey(Customer, related_name="buyings", blank=False)
|
|
date = models.DateTimeField(_('date'), auto_now=True)
|
|
is_validated = models.BooleanField(_('is validated'), default=False)
|
|
|
|
class Meta:
|
|
verbose_name = _("selling")
|
|
|
|
def __str__(self):
|
|
return "Selling: %d x %s (%f) for %s" % (self.quantity, self.label,
|
|
self.quantity*self.unit_price, self.customer.user.get_display_name())
|
|
|
|
def save(self, *args, **kwargs):
|
|
self.full_clean()
|
|
if not self.is_validated:
|
|
self.customer.amount -= self.quantity * self.unit_price
|
|
self.customer.save()
|
|
self.is_validated = True
|
|
super(Selling, self).save(*args, **kwargs)
|
|
|
|
class Permanency(models.Model):
|
|
"""
|
|
This class aims at storing a traceability of who was barman where and when
|
|
"""
|
|
user = models.ForeignKey(User, related_name="permanencies")
|
|
counter = models.ForeignKey(Counter, related_name="permanencies")
|
|
start = models.DateTimeField(_('start date'))
|
|
end = models.DateTimeField(_('end date'))
|
|
|
|
class Meta:
|
|
verbose_name = _("permanency")
|
|
|
|
def __str__(self):
|
|
return "%s in %s from %s to %s" % (self.user, self.counter,
|
|
self.start.strftime("%Y-%m-%d %H:%M:%S"), self.end.strftime("%Y-%m-%d %H:%M:%S"))
|
|
|
|
|