oxpanel/oxpanel25
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

vpn integration

Browse Source
This commit is contained in:
Nell
2026-04-11 22:07:59 +02:00
parent c4d27e9842
commit 00ac38d126
47 changed files with 945 additions and 749 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/torrent/consumers.py
+36 -32
View File
@@ -1,11 +1,8 @@
from django.db.models import Q
from django.db.models.functions import Coalesce
from channels.generic.websocket import AsyncJsonWebsocketConsumer
from typing import Optional, Union
import asyncio
from django.db.models import Q
from user.models import User
from .models import Torrent
@@ -14,17 +11,19 @@ class TorrentEventConsumer(AsyncJsonWebsocketConsumer):
super().__init__(*args, **kwargs)
self.channel_groups = set()
self.user: Optional[User] = None
self.follow_user: Optional[User] = None
self.user: User | None = None
self.follow_user: User | None = None
async def connect(self):
self.user = self.scope['user']
self.user = self.scope["user"]
if not self.user.is_authenticated:
await self.close()
return
self.follow_user = self.user
await self.channel_layer.group_add(f"user_{self.follow_user.id}", self.channel_name)
await self.channel_layer.group_add(
f"user_{self.follow_user.id}", self.channel_name
)
# user_id = int(self.scope['url_route']["kwargs"]["user_id"])
# if user_id == self.user.id:
@@ -53,47 +52,52 @@ class TorrentEventConsumer(AsyncJsonWebsocketConsumer):
print("call websocket not supported", content)
async def change_follow_user(self, user_id):
await self.channel_layer.group_discard(f"user_{self.follow_user.id}", self.channel_name)
await self.channel_layer.group_discard(
f"user_{self.follow_user.id}", self.channel_name
)
if user_id == self.user.id:
self.follow_user = self.user
await self.channel_layer.group_add(f"user_{self.follow_user.id}", self.channel_name)
await self.channel_layer.group_add(
f"user_{self.follow_user.id}", self.channel_name
)
return self.follow_user
elif await self.user.friends.filter(id=user_id).aexists():
self.follow_user = await User.objects.filter(id=user_id).aget()
await self.channel_layer.group_add(f"user_{self.follow_user.id}", self.channel_name)
await self.channel_layer.group_add(
f"user_{self.follow_user.id}", self.channel_name
)
return self.follow_user
else:
return None
async def transmission_data_updated(self, datas):
torrent_stats = datas["data"]
qs = (Torrent.objects
.filter(Q(user_id=self.follow_user.id) | Q(shared_users=self.follow_user.id))
.values_list("id", flat=True).distinct())
qs = (
Torrent.objects.filter(
Q(user_id=self.follow_user.id) | Q(shared_users=self.follow_user.id)
)
.values_list("id", flat=True)
.distinct()
)
torrent_ids = [i async for i in qs]
for hash_string, data in torrent_stats.items():
if hash_string in torrent_ids:
await self.send_json({
"context": "transmission_data_updated",
"data": data
})
await self.send_json(
{"context": "transmission_data_updated", "data": data}
)
async def add_torrent(self, data):
await self.send_json({
"context": "add_torrent",
"torrent_id": data["data"]
})
await self.send_json({"context": "add_torrent", "torrent_id": data["data"]})
async def remove_torrent(self, data):
await self.send_json({
"context": "remove_torrent",
"torrent_id": data["data"]
})
await self.send_json({"context": "remove_torrent", "torrent_id": data["data"]})
async def update_torrent(self, data):
await self.send_json({
"context": "update_torrent",
"torrent_id": data["data"]["torrent_id"],
"updated_fields": data["data"]["updated_fields"]
})
await self.send_json(
{
"context": "update_torrent",
"torrent_id": data["data"]["torrent_id"],
"updated_fields": data["data"]["updated_fields"],
}
)