PTV/v6d3music
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: PTV:abae781b5249e39b35e3910fe9df4899de41722e
Branches Tags
PTV:master
..
compare: PTV:6b4c8aa8fa3e0ad15cd96563e7fe687074d5e813
Branches Tags
PTV:master

No commits in common. "abae781b5249e39b35e3910fe9df4899de41722e" and "6b4c8aa8fa3e0ad15cd96563e7fe687074d5e813" have entirely different histories.
abae781b52 ... 6b4c8aa8fa

8 changed files with 36 additions and 75 deletions
Show Stats Expand all files Collapse all files

2
base.requirements.txt
View File

@ -1,4 +1,4 @@
aiohttp>=3.7.4,<4
discord.py[voice]~=2.2.2
discord.py[voice]~=2.2.1
yt-dlp~=2023.2.17
typing_extensions~=4.4.0

2
requirements.txt
View File

@ -1,6 +1,6 @@
ptvp35 @ git+https://gitea.parrrate.ru/PTV/ptvp35.git@e760fca39e2070b9959aeb95b53e59e895f1ad57
v6d0auth @ git+https://gitea.parrrate.ru/PTV/v6d0auth.git@c718d4d1422945a756213d22d9e26aa24babe0f6
v6d1tokens @ git+https://gitea.parrrate.ru/PTV/v6d1tokens.git@9ada50f111bd6e9a49c9c6683fa7504fee030056
v6d2ctx @ git+https://gitea.parrrate.ru/PTV/v6d2ctx.git@226bf1b6ada0c217408590abf69379d5f44a7972
v6d2ctx @ git+https://gitea.parrrate.ru/PTV/v6d2ctx.git@c9f3f5ac5c7feb2165fc4fae4eb998a0fe4f5f00
rainbowadn @ git+https://gitea.parrrate.ru/PTV/rainbowadn.git@fc1d11f4b53ac4653ffac1bbcad130855e1b7f10
adaas @ git+https://gitea.parrrate.ru/PTV/adaas.git@8093665489901098f92d5a4001f1782dab6ddcf9

18
v6d3music/commands.py
View File

@ -3,7 +3,7 @@ from typing import Callable
import discord
from v6d2ctx.at_of import AtOf
from v6d2ctx.at_of import *
from v6d2ctx.context import *
from v6d3music.core.default_effects import *
from v6d3music.core.mainservice import *
@ -64,10 +64,6 @@ presets: {shlex.join(allowed_presets)}
queue.append(audio)
await ctx.reply("done")
@at("cancel")
async def cancel(ctx: Context, _args: list[str]) -> None:
mainservice.cancel_loading(ctx)
@at("skip")
async def skip(ctx: Context, args: list[str]) -> None:
await catch(
@ -279,18 +275,6 @@ presets: {shlex.join(allowed_presets)}
or "no queue"
)
@at("np")
@at("cp")
@at("nowplaying")
@at("playing")
@at("//1")
@at("queue1")
async def queue1(ctx: Context, _args: list[str]) -> None:
await ctx.long(
(await (await mainservice.context(ctx, create=True, force_play=False).queue()).format(1)).strip()
or "no queue"
)
@at("swap")
async def swap(ctx: Context, args: list[str]) -> None:
await catch(

20
v6d3music/core/mainservice.py
View File

@ -1,7 +1,7 @@
import asyncio
import traceback
from contextlib import AsyncExitStack
from typing import AsyncIterable, Callable, TypeVar
from typing import AsyncIterable, TypeVar
import discord
@ -22,7 +22,6 @@ from v6d3music.core.ystate import *
from v6d3music.core.ytaservicing import *
from v6d3music.core.ytaudio import *
from v6d3music.processing.pool import *
from v6d3music.utils.assert_admin import assert_admin
from v6d3music.utils.argctx import *
__all__ = ('MainService', 'MainMode', 'MainContext', 'MainEvent')
@ -66,7 +65,6 @@ class MainService:
self.restore_lock = asyncio.Lock()
self.__events: SendableEvents[MainEvent] = events
self.__pool_events: SendableEvents[PoolEvent] = _PMSendable(self.__events)
self.__ystates: dict[discord.Guild, YState] = {}
def register_instrumentation(self):
self.targets.register_type(v6d3music.processing.pool.UnitJob, 'run', Async)
@ -233,22 +231,8 @@ class MainService:
async def yt_audios(self, ctx: Context, args: list[str]) -> AsyncIterable[YTAudio]:
assert ctx.guild is not None
argctx = ArgCtx(self.defaulteffects.get(ctx.guild.id), args)
ystate = YState(self.__servicing, self.__pool, ctx, argctx.sources)
self.__ystates[ctx.guild] = ystate
try:
async for audio in ystate.iterate():
async for audio in YState(self.__servicing, self.__pool, ctx, argctx.sources).iterate():
yield audio
finally:
del self.__ystates[ctx.guild]
def cancel_loading(self, ctx: Context) -> None:
assert ctx.guild is not None
ystate = self.__ystates.get(ctx.guild)
if ystate is None:
return
if ystate.ctx.member != ctx.member:
assert_admin(ctx.member)
ystate.cancel()
def pool_json(self) -> ResponseType:
return self.__pool.json()

2
v6d3music/core/queueaudio.py
View File

@ -16,7 +16,7 @@ from v6d3music.utils.fill import *
__all__ = ('QueueAudio',)
PRE_SET_LENGTH = 6
PRE_SET_LENGTH = 24
class QueueAudio(discord.AudioSource):

61
v6d3music/core/ystate.py
View File

@ -3,15 +3,16 @@ from collections import deque
from contextlib import AsyncExitStack
from typing import AsyncIterable, Iterable
from v6d2ctx.context import *
from v6d2ctx.integration.responsetype import *
from v6d2ctx.context import *
from v6d3music.core.create_ytaudio import *
from v6d3music.core.ytaservicing import *
from v6d3music.core.ytaudio import *
from v6d3music.processing.pool import *
from v6d3music.utils.argctx import *
__all__ = ("YState",)
__all__ = ('YState',)
class _Stop:
@ -47,9 +48,7 @@ class YState:
try:
return await future
except OSError as e:
raise Explicit(
"extraction error\nunknown ytdl error (probably due to video being unavailable, e.g. because of regional restrictions)"
) from e
raise Explicit('extraction error\nunknown ytdl error (probably due to video being unavailable, e.g. because of regional restrictions)') from e
finally:
self.results.task_done()
@ -83,18 +82,12 @@ class YState:
else:
return None
def cancel(self) -> None:
self.entries.clear()
self.playlists.clear()
self.sources.clear()
self.results.put_nowait(_Stop())
class YStream(JobUnit):
def __init__(self, state: YState) -> None:
self.state = state
self.__running = False
self.__details: dict[str, ResponseType] = {"status": "stopped"}
self.__details: dict[str, ResponseType] = {'status': 'stopped'}
def _unpack_playlists(self) -> None:
while self.state.playlists and self.state.playlists[0].done():
@ -109,7 +102,7 @@ class YStream(JobUnit):
async def _run(self, context: JobContext, /) -> JobUnit | None:
if self.state.empty_processing():
self.__details = {"status": "stopping"}
self.__details = {'status': 'stopping'}
if self.state.results.empty():
self.state.results.put_nowait(_Stop())
return None
@ -117,20 +110,20 @@ class YStream(JobUnit):
entry = self.state.entries.popleft()
audiotask: asyncio.Future[YTAudio | None]
if isinstance(entry, BaseException):
self._set_details(context, {"status": "breaking downstream audio creation"})
self._set_details(context, {'status': 'breaking downstream audio creation'})
audiotask = asyncio.Future()
audiotask.set_exception(entry)
else:
self._set_details(
context,
{
"status": "creating audio",
"info": cast_to_response(entry.info),
"effects": entry.effects,
"already_read": entry.already_read,
"tor": entry.tor,
"ignore": entry.ignore,
},
'status': 'creating audio',
'info': cast_to_response(entry.info),
'effects': entry.effects,
'already_read': entry.already_read,
'tor': entry.tor,
'ignore': entry.ignore,
}
)
audiotask = asyncio.create_task(self.state.result(entry))
self.state.results.put_nowait(audiotask)
@ -140,20 +133,20 @@ class YStream(JobUnit):
self.state.entries.clear()
self.state.playlists.clear()
self.state.sources.clear()
self._set_details(context, {"status": "rescheduling self from entries"})
self._set_details(context, {'status': 'rescheduling self from entries'})
return self
elif self.state.sources:
source = self.state.sources.popleft()
self._set_details(
context,
{
"status": "parsing playlist",
"url": source.url,
"effects": source.effects,
"already_read": source.already_read,
"tor": source.tor,
"ignore": source.ignore,
},
'status': 'parsing playlist',
'url': source.url,
'effects': source.effects,
'already_read': source.already_read,
'tor': source.tor,
'ignore': source.ignore,
}
)
playlisttask = asyncio.create_task(self.state.playlist(source))
self.state.playlists.append(playlisttask)
@ -166,13 +159,13 @@ class YStream(JobUnit):
self._unpack_playlists()
rescheduled = self.state.descheduled
self.state.descheduled = 0
self._set_details(context, {"status": "rescheduling others", "rescheduling": rescheduled})
self._set_details(context, {'status': 'rescheduling others', 'rescheduling': rescheduled})
for _ in range(rescheduled):
await self.state.es.enter_async_context(YStream(self.state).at(self.state.pool))
self._set_details(context, {"status": "rescheduling self from sources"})
self._set_details(context, {'status': 'rescheduling self from sources'})
return self
else:
self._set_details(context, {"status": "descheduling"})
self._set_details(context, {'status': 'descheduling'})
self.state.descheduled += 1
return None
@ -186,7 +179,7 @@ class YStream(JobUnit):
return await self._run(context)
finally:
self.__running = False
self.__details = {"status": "stopped"}
self.__details = {'status': 'stopped'}
def json(self) -> ResponseType:
return {"type": "ystream", "details": self.__details, "running": self.__running}
return {'type': 'ystream', 'details': self.__details, 'running': self.__running}

2
v6d3music/main.py
View File

@ -15,7 +15,7 @@ from v6d1tokens.client import *
from v6d2ctx.handle_content import *
from v6d2ctx.integration.event import *
from v6d2ctx.integration.targets import *
from v6d2ctx.pain import ABlockMonitor, ALog
from v6d2ctx.pain import ABlockMonitor
from v6d2ctx.serve import *
from v6d3music.api import *
from v6d3music.app import *

2
v6d3music/utils/assert_admin.py
View File

@ -6,7 +6,7 @@ __all__ = ('assert_admin',)
def assert_admin(member: discord.Member | None):
assert isinstance(member, discord.Member)
assert member is not None
permissions: discord.Permissions = member.guild_permissions
if not permissions.administrator:
raise Explicit('not an administrator')