Improve and fix user accounts

counter/models.py

@@ -5,8 +5,7 @@ from django.conf import settings
 from django.core.urlresolvers import reverse
 from django.forms import ValidationError
 
-from datetime import timedelta, datetime
-from pytz import timezone
+from datetime import timedelta
 import random
 import string
 
@@ -188,7 +187,7 @@ class Counter(models.Model):
 
     def get_random_barman(self):
         bl = self.get_barmen_list()
-        return bl[randrange(0, len(bl))]
+        return bl[random.randrange(0, len(bl))]
 
     def is_open(self):
         response = False
@@ -220,12 +219,20 @@ class Refilling(models.Model):
     def __str__(self):
         return "Refilling: %.2f for %s" % (self.amount, self.customer.user.get_display_name())
 
+    def is_owned_by(self, user):
+        return user.can_edit(self.counter) and self.payment_method != "CARD"
+
     # def get_absolute_url(self):
     #     return reverse('counter:details', kwargs={'counter_id': self.id})
 
+    def delete(self, *args, **kwargs):
+        self.customer.amount -= self.amount
+        self.customer.save()
+        super(Refilling, self).delete(*args, **kwargs)
+
     def save(self, *args, **kwargs):
         if not self.date:
-            self.date = datetime.now().replace(tzinfo=timezone(settings.TIME_ZONE))
+            self.date = timezone.now()
         self.full_clean()
         if not self.is_validated:
             self.customer.amount += self.amount
@@ -257,9 +264,17 @@ class Selling(models.Model):
         return "Selling: %d x %s (%f) for %s" % (self.quantity, self.label,
                 self.quantity*self.unit_price, self.customer.user.get_display_name())
 
+    def is_owned_by(self, user):
+        return user.can_edit(self.counter) and self.payment_method != "CARD"
+
+    def delete(self, *args, **kwargs):
+        self.customer.amount += self.quantity * self.unit_price
+        self.customer.save()
+        super(Selling, self).delete(*args, **kwargs)
+
     def save(self, *args, **kwargs):
         if not self.date:
-            self.date = datetime.now().replace(tzinfo=timezone(settings.TIME_ZONE))
+            self.date = timezone.now()
         self.full_clean()
         if not self.is_validated:
             self.customer.amount -= self.quantity * self.unit_price

counter/urls.py

@@ -17,6 +17,8 @@ urlpatterns = [
     url(r'^admin/producttype/list$', ProductTypeListView.as_view(), name='producttype_list'),
     url(r'^admin/producttype/create$', ProductTypeCreateView.as_view(), name='new_producttype'),
     url(r'^admin/producttype/(?P<type_id>[0-9]+)$', ProductTypeEditView.as_view(), name='producttype_edit'),
+    url(r'^admin/selling/(?P<selling_id>[0-9]+)/delete$', SellingDeleteView.as_view(), name='selling_delete'),
+    url(r'^admin/refilling/(?P<refilling_id>[0-9]+)/delete$', RefillingDeleteView.as_view(), name='refilling_delete'),
 ]
 
 

counter/views.py

@@ -410,3 +410,26 @@ class ProductEditView(CanEditPropMixin, UpdateView):
     template_name = 'core/edit.jinja'
     # TODO: add management of the 'counters' ForeignKey
 
+
+class RefillingDeleteView(CanEditPropMixin, DeleteView):
+    """
+    Delete a refilling (for the admins)
+    """
+    model = Refilling
+    pk_url_kwarg = "refilling_id"
+    template_name = 'core/delete_confirm.jinja'
+
+    def get_success_url(self):
+        return reverse_lazy('core:user_account', kwargs={'user_id': self.object.customer.user.id})
+
+class SellingDeleteView(CanEditPropMixin, DeleteView):
+    """
+    Delete a selling (for the admins)
+    """
+    model = Selling
+    pk_url_kwarg = "selling_id"
+    template_name = 'core/delete_confirm.jinja'
+
+    def get_success_url(self):
+        return reverse_lazy('core:user_account', kwargs={'user_id': self.object.customer.user.id})
+