Skip to content

Commit c948beb

Browse files
committed
Add support to change log url prefix
1 parent c982106 commit c948beb

File tree

3 files changed

+16
-4
lines changed

3 files changed

+16
-4
lines changed

cogs/modmail.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import asyncio
22
import re
3+
import os
34
from datetime import datetime
45
from typing import Optional, Union
56
from types import SimpleNamespace as param
@@ -381,7 +382,11 @@ def format_log_embeds(self, logs, avatar_url):
381382

382383
created_at = parser.parse(entry['created_at'])
383384

384-
log_url = self.bot.config.log_url.strip('/') + f'/logs/{key}'
385+
prefix = os.getenv('LOG_URL_PREFIX', '/logs')
386+
prefix = ''
387+
388+
389+
log_url = self.bot.config.log_url.strip('/') + f'{prefix}/{key}'
385390

386391
username = entry['recipient']['name'] + '#'
387392
username += entry['recipient']['discriminator']

core/clients.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import os
12
import logging
23
import secrets
34
from datetime import datetime
@@ -14,6 +15,9 @@
1415

1516
logger = logging.getLogger('Modmail')
1617

18+
prefix = os.getenv('LOG_URL_PREFIX', '/logs')
19+
if prefix == 'NONE':
20+
prefix = ''
1721

1822
class ApiClient:
1923
"""
@@ -288,7 +292,7 @@ async def get_log(self, channel_id):
288292

289293
async def get_log_link(self, channel_id):
290294
doc = await self.get_log(channel_id)
291-
return f"{self.bot.config.log_url.strip('/')}/logs/{doc['key']}"
295+
return f"{self.bot.config.log_url.strip('/')}{prefix}/{doc['key']}"
292296

293297
async def create_log_entry(self, recipient, channel, creator):
294298
key = secrets.token_hex(6)
@@ -318,7 +322,7 @@ async def create_log_entry(self, recipient, channel, creator):
318322
'messages': []
319323
})
320324

321-
return f"{self.bot.config.log_url.strip('/')}/logs/{key}"
325+
return f"{self.bot.config.log_url.strip('/')}{prefix}/{key}"
322326

323327
async def get_config(self):
324328
conf = await self.db.config.find_one({'bot_id': self.bot.user.id})

core/thread.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import asyncio
22
import logging
33
import re
4+
import os
45
import string
56
import typing
67
from datetime import datetime, timedelta
@@ -221,7 +222,9 @@ async def _close(self, closer, silent=False, delete_channel=True,
221222
})
222223

223224
if log_data is not None and isinstance(log_data, dict):
224-
log_url = f"{self.bot.config.log_url.strip('/')}/logs/{log_data['key']}"
225+
prefix = os.getenv('LOG_URL_PREFIX', '/logs')
226+
if prefix == 'NONE': prefix = ''
227+
log_url = f"{self.bot.config.log_url.strip('/')}{prefix}/{log_data['key']}"
225228

226229
if log_data['messages']:
227230
content = str(log_data['messages'][0]['content'])

0 commit comments

Comments
 (0)