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
91
92
93
94
95
96
97
98
99
100
101
102
103
|
import logging
from flask import jsonify, request
from schema import Optional, Schema
from pysite.base_route import APIView
from pysite.constants import ValidationTypes
from pysite.decorators import api_key, api_params
from pysite.mixins import DBMixin
SCHEMA = Schema([
{
"user_id": str,
"roles": [str],
"username": str,
"discriminator": str
}
])
DELETE_SCHEMA = Schema([
{
"user_id": str,
Optional("roles"): [str],
Optional("username"): str,
Optional("discriminator"): str
}
])
class UserView(APIView, DBMixin):
path = "/user"
name = "api.bot.user"
table_name = "users"
oauth_table_name = "oauth_data"
@api_key
@api_params(schema=SCHEMA, validation_type=ValidationTypes.json)
def post(self, data):
logging.getLogger(__name__).debug(f"Size of request: {len(request.data)} bytes")
deletions = 0
oauth_deletions = 0
user_ids = [user["user_id"] for user in data]
all_users = self.db.run(self.db.query(self.table_name), coerce=list)
for user in all_users:
if user["user_id"] not in user_ids:
self.db.delete(self.table_name, user["user_id"], durability="soft")
deletions += 1
all_oauth_data = self.db.run(self.db.query(self.oauth_table_name), coerce=list)
for item in all_oauth_data:
if item["snowflake"] not in user_ids:
self.db.delete(self.oauth_table_name, item["id"], durability="soft")
oauth_deletions += 1
del user_ids
changes = self.db.insert(
self.table_name, *data,
conflict="update",
durability="soft"
)
self.db.sync(self.table_name)
changes["deleted"] = deletions
changes["deleted_oauth"] = oauth_deletions
return jsonify(changes) # pragma: no cover
@api_key
@api_params(schema=SCHEMA, validation_type=ValidationTypes.json)
def put(self, data):
changes = self.db.insert(
self.table_name, *data,
conflict="update"
)
return jsonify(changes) # pragma: no cover
@api_key
@api_params(schema=DELETE_SCHEMA, validation_type=ValidationTypes.json)
def delete(self, data):
user_ids = [user["user_id"] for user in data]
changes = self.db.run(
self.db.query(self.table_name)
.get_all(*user_ids)
.delete()
)
oauth_deletions = self.db.run(
self.db.query(self.oauth_table_name)
.get_all(*user_ids, index="snowflake")
.delete()
).get("deleted", 0)
changes["deleted_oauth"] = oauth_deletions
return jsonify(changes) # pragma: no cover
|
