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
|
from datetime import datetime
from sqlalchemy import select, delete
from sqlalchemy.ext.asyncio import AsyncSession
from app.models import Device, Subscription, Event, Delivery
from app.schemas import DeviceRegisterIn, SubscriptionUpsertIn
async def upsert_device(db: AsyncSession, payload: DeviceRegisterIn) -> Device:
stmt = select(Device).where(
Device.apns_token == payload.apns_token,
Device.bundle_id == payload.bundle_id,
)
result = await db.execute(stmt)
device = result.scalar_one_or_none()
if device is None:
device = Device(
user_id=payload.user_id,
apns_token=payload.apns_token,
apns_env=payload.apns_env,
bundle_id=payload.bundle_id,
platform=payload.platform,
app_version=payload.app_version,
device_name=payload.device_name,
is_enabled=True,
last_seen_at=datetime.utcnow(),
)
db.add(device)
else:
device.user_id = payload.user_id
device.apns_env = payload.apns_env
device.platform = payload.platform
device.app_version = payload.app_version
device.device_name = payload.device_name
device.is_enabled = True
device.last_seen_at = datetime.utcnow()
await db.commit()
await db.refresh(device)
return device
async def replace_subscriptions(db: AsyncSession, payload: SubscriptionUpsertIn) -> None:
await db.execute(delete(Subscription).where(Subscription.device_id == payload.device_id))
for item in payload.subscriptions:
db.add(
Subscription(
device_id=payload.device_id,
source_type=item.source_type,
source_id=item.source_id,
event_type=item.event_type,
is_enabled=item.is_enabled,
)
)
await db.commit()
|
