47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
from settings.user_settings import UserSettings, UserSetting
|
|
from settings.setting_types import Boolean
|
|
|
|
from modules.reminders.reminder import Reminder
|
|
from modules.reminders.data import reminders
|
|
|
|
from .utils import create_remainder, remainder_content
|
|
|
|
|
|
@UserSettings.attach_setting
|
|
class topgg_vote_remainder(Boolean, UserSetting):
|
|
attr_name = 'vote_remainder'
|
|
_data_column = 'remaind_upvote'
|
|
|
|
_default = True
|
|
|
|
display_name = 'Upvote Reminder'
|
|
desc = "Turn on/off DM Reminders to Upvote me."
|
|
long_desc = ("Enable or disable DM boost reminders.",)
|
|
|
|
@property
|
|
def success_response(self):
|
|
if self.value:
|
|
# Check if reminder is already running
|
|
create_remainder(self.id)
|
|
|
|
return (
|
|
" I will send you boost reminders.\n\n"
|
|
"`Please make sure your DMs are open.`"
|
|
)
|
|
else:
|
|
# Check if reminder is already running and get its id
|
|
r = reminders.select_one_where(
|
|
userid=self.id,
|
|
select_columns='reminderid',
|
|
content=remainder_content,
|
|
_extra="ORDER BY remind_at DESC LIMIT 1"
|
|
)
|
|
|
|
# Cancel and delete Remainder if already running
|
|
if r:
|
|
Reminder.delete(r['reminderid'])
|
|
|
|
return (
|
|
" I won't send you boost reminders."
|
|
)
|