1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
from django.db import models
from django.contrib.auth.models import User
from django.shortcuts import Http404, get_object_or_404
from datetime import datetime
class Restaurant(models.Model):
name = models.CharField(max_length=50)
users = models.ManyToManyField(User)
wordpress_url = models.URLField(max_length=50, unique=True)
woocommerce_consumer_key = models.CharField(max_length=50)
woocommerce_consumer_secret = models.CharField(max_length=50)
woocommerce_webhook_secret = models.CharField(max_length=50)
@classmethod
def get_user_restaurants(cls, user: User):
return cls.objects.filter(users=user)
@classmethod
def get_user_restaurant_or_404(cls, pk, user: User):
return get_object_or_404(cls, pk=pk, users=user)
def __str__(self):
return self.name
class Order(models.Model):
WP_STATES = (
('pending', 'Oczekujace'),
('processing', 'Przetwarzane'),
('on-hold', 'Wstrzymane'),
('completed', 'Zakonczone'),
('cancelled', 'Anulowane'),
('refunded', 'Zwrocone'),
('failed', 'Nie powiodlo sie'),
('trash', 'Usuniete')
)
wp_id = models.IntegerField(editable=False)
wp_status = models.CharField(max_length=30, choices=WP_STATES)
wp_order_key = models.CharField(max_length=50, editable=False)
date_created = models.DateField(editable=False)
date_modified = models.DateField()
line_items = models.JSONField()
billing = models.JSONField()
restaurant = models.ForeignKey(Restaurant, on_delete=models.CASCADE)
can_display = models.BooleanField(default=True)
@classmethod
def update_or_create_from_response(cls, response, restaurant_model):
try:
obj = cls.objects.update_or_create(
wp_id=response['id'],
wp_order_key=response['order_key'],
restaurant=restaurant_model,
defaults={
'wp_status': response['status'],
'date_created': datetime.strptime(response['date_created'], '%Y-%m-%dT%H:%M:%S'),
'date_modified': datetime.strptime(response['date_modified'], '%Y-%m-%dT%H:%M:%S'),
'line_items': response['line_items'],
'billing': response['billing'],
}
)
return obj
except KeyError:
return None
@classmethod
def create_from_response_disable_view(cls, response, restaurant_model):
obj, _ = cls.update_or_create_from_response(response, restaurant_model)
obj.can_display = False
obj.save()
return obj
@classmethod
def get_order(cls, pk, user: User):
try:
return cls.objects.get(
pk=pk,
can_display=True,
restaurant__users=user
)
except cls.DoesNotExist:
raise Http404
def __str__(self):
return f'{self.wp_order_key} - {self.date_created}'
|
