mirror of
https://github.com/ethanrusz/youtube-url-corrector.git
synced 2024-12-13 19:57:47 -05:00
Merge pull request #3 from ethanrusz/url-changes
Update Docker URL configuration
This commit is contained in:
commit
9121816fc6
2 changed files with 22 additions and 19 deletions
|
@ -10,6 +10,6 @@ services:
|
|||
image: git.beans.team/em/yuc:latest
|
||||
environment:
|
||||
- DISCORD_TOKEN=your_discord_bot_token
|
||||
- PIPED_URL=https://your.piped.url/watch?v=
|
||||
- PIPED_URL=https://your.piped.url # Do not append /watch?v=
|
||||
restart: unless-stopped
|
||||
```
|
||||
|
|
39
bot.py
39
bot.py
|
@ -10,10 +10,10 @@ intents = discord.Intents.default()
|
|||
intents.members = True
|
||||
intents.message_content = True
|
||||
|
||||
bot = commands.Bot(command_prefix='?', intents=intents)
|
||||
bot = commands.Bot(command_prefix="?", intents=intents)
|
||||
|
||||
PIPED_URL = os.getenv('PIPED_URL')
|
||||
DISCORD_TOKEN = os.getenv('DISCORD_TOKEN')
|
||||
PIPED_URL = os.getenv("PIPED_URL") + "/watch?v="
|
||||
DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
|
||||
|
||||
|
||||
@bot.event
|
||||
|
@ -24,7 +24,8 @@ async def on_ready():
|
|||
@bot.event
|
||||
async def on_message(message):
|
||||
regex = re.compile(
|
||||
r'https://(music\.)?(www\.)?youtu(be)?\.[a-zA-Z0-9()]{1,6}\b([-a-zA-Z0-9()@:%_+.~#?&/=]*)')
|
||||
r"https://(music\.)?(www\.)?youtu(be)?\.[a-zA-Z0-9()]{1,6}\b([-a-zA-Z0-9()@:%_+.~#?&/=]*)"
|
||||
)
|
||||
|
||||
result = regex.search(message.content)
|
||||
if result is not None:
|
||||
|
@ -32,28 +33,30 @@ async def on_message(message):
|
|||
video_id = await get_youtube_id(youtube_url)
|
||||
|
||||
if video_id is not None:
|
||||
await message.reply(f"I think you meant {PIPED_URL}{video_id}.", mention_author=False)
|
||||
await message.reply(
|
||||
f"I think you meant {PIPED_URL}{video_id}.", mention_author=False
|
||||
)
|
||||
|
||||
|
||||
async def get_youtube_id(url: str, ignore_playlist=True) -> str:
|
||||
query = urlparse(url)
|
||||
if query.hostname == 'youtu.be':
|
||||
if query.hostname == "youtu.be":
|
||||
return query.path[1:]
|
||||
if query.hostname in {'www.youtube.com', 'youtube.com', 'music.youtube.com'}:
|
||||
if query.hostname in {"www.youtube.com", "youtube.com", "music.youtube.com"}:
|
||||
if not ignore_playlist:
|
||||
# use case: get playlist id not current video in playlist
|
||||
with suppress(KeyError):
|
||||
return parse_qs(query.query)['list'][0]
|
||||
if query.path == '/watch':
|
||||
return parse_qs(query.query)['v'][0]
|
||||
if query.path[:7] == '/watch/':
|
||||
return query.path.split('/')[1]
|
||||
if query.path[:7] == '/embed/':
|
||||
return query.path.split('/')[2]
|
||||
if query.path[:3] == '/v/':
|
||||
return query.path.split('/')[2]
|
||||
if query.path[:8] == '/shorts/':
|
||||
return query.path.split('/')[1]
|
||||
return parse_qs(query.query)["list"][0]
|
||||
if query.path == "/watch":
|
||||
return parse_qs(query.query)["v"][0]
|
||||
if query.path[:7] == "/watch/":
|
||||
return query.path.split("/")[1]
|
||||
if query.path[:7] == "/embed/":
|
||||
return query.path.split("/")[2]
|
||||
if query.path[:3] == "/v/":
|
||||
return query.path.split("/")[2]
|
||||
if query.path[:8] == "/shorts/":
|
||||
return query.path.split("/")[1]
|
||||
|
||||
|
||||
bot.run(DISCORD_TOKEN)
|
||||
|
|
Loading…
Reference in a new issue