def get_flag_by_notify(self, notify):
if notify.user.profile.check_role(Role.BANNED):
return False
if notify.repeat:
if notify.crontab:
timezone = notify.user.profile.city.timezone.name
localized_datetime = localize_datetime(remove_tz(self.dt_now),
timezone)
entry = CronTab(notify.crontab)
prev_seconds_delta = -entry.previous(localized_datetime,
default_utc=True)
return prev_seconds_delta <= 60
else:
datetime1 = datetime.combine(date.min,
remove_tz(notify.date).time())
datetime2 = datetime.combine(date.min, self.dt_now.time())
delta_time = datetime1 - datetime2 + timedelta(minutes=1)
return delta_time.seconds <= 60
else:
delta_time = remove_tz(
notify.date) - self.dt_now + timedelta(minutes=1)
return delta_time.days == 0 and delta_time.seconds <= 60
def stop_if_need(self):
self.get_server_info()
# Если сервак онлайн и нет игроков
if self.server_info['online'] and not self.server_info['players']:
obj, created = Service.objects.get_or_create(name=f'stop_{self._get_service_name()}')
# Создание событие. Уведомление, что мы скоро всё отрубим
if created:
message = f"Если никто не зайдёт на сервак по майну {self.get_version()}, то через полчаса я его остановлю"
users_notify = Profile.objects.filter(groups__name=Role.MINECRAFT_NOTIFY.name)
for user in users_notify:
bot = get_bot_by_platform(user.get_platform_enum())
bot.parse_and_send_msgs_thread(message, user.user_id)
# Если событие уже было создано, значит пора отрубать
else:
update_datetime = obj.update_datetime
delta_seconds = (datetime.utcnow() - remove_tz(update_datetime)).seconds
if delta_seconds <= 1800 + 100:
obj.delete()
Service.objects.get_or_create(self._get_service_name())
self.stop(send_notify=False)
message = f"Вырубаю майн {self.get_version()}"
users_notify = Profile.objects.filter(groups__name=Role.MINECRAFT_NOTIFY.name)
for user in users_notify:
bot = get_bot_by_platform(user.get_platform_enum())
bot.parse_and_send_msgs_thread(message, user.user_id)
else:
obj.delete()
# Эта ветка нужна, чтобы вручную вырубленные серверы не провоцировали при последующем старте отключение в 0/30 минут
else:
Service.objects.filter(name=f'stop_{self._get_service_name()}').delete()
def get_time(arg1, arg2, timezone=None):
exact_datetime_flag = True
if arg1 in DELTA_WEEKDAY:
exact_datetime_flag = False
arg1 = (datetime.today().date() +
timedelta(days=DELTA_WEEKDAY[arg1])).strftime("%d.%m.%Y")
if arg1 in WEEK_TRANSLATOR:
exact_datetime_flag = False
delta_days = WEEK_TRANSLATOR[arg1] - datetime.today().isoweekday()
if delta_days <= 0:
delta_days += 7
arg1 = (datetime.today().date() +
timedelta(days=delta_days)).strftime("%d.%m.%Y")
default_datetime = datetime.utcnow().replace(
hour=10, minute=0, second=0, microsecond=0) # + timedelta(days=1)
# ToDo: зачем? тут баг из-за которого время ебошится неправильно при завтра/послезавтра
default_datetime = remove_tz(
normalize_datetime(default_datetime, tz=timezone.name))
try:
if arg1.count('.') == 1:
arg1 = f"{arg1}.{default_datetime.year}"
date_str = f"{arg1} {arg2}"
return parser.parse(date_str, default=default_datetime,
dayfirst=True), 2, exact_datetime_flag
except dateutil.parser._parser.ParserError:
try:
exact_datetime_flag = False
return parser.parse(arg1, default=default_datetime,
dayfirst=True), 1, exact_datetime_flag
except dateutil.parser._parser.ParserError:
return None, None, None
def get_notifies_from_object(notifies_obj, timezone, print_username=False):
if len(notifies_obj) == 0:
raise PWarning("Нет напоминаний")
result = ""
for notify in notifies_obj:
if notify.date:
notify_datetime = localize_datetime(remove_tz(notify.date), timezone)
else:
notify_datetime = notify.crontab
if print_username:
result += f"{notify.user}\n"
if notify.repeat:
if notify.crontab:
result += f"{notify_datetime} - Постоянное"
else:
result += f"{str(notify_datetime.strftime('%H:%M'))} - Постоянное"
else:
result += f"{str(notify_datetime.strftime('%d.%m.%Y %H:%M'))} - Постоянное"
if notify.chat:
result += f" (Конфа - {notify.chat.name})"
result += f"\n{notify.text}\n\n"
result_without_mentions = result.replace('@', '@_')
return result_without_mentions
def get_avg_price_data(data, tariff):
tariff_map = {
'econom': 0,
'comfort': 1,
'comfortplus': 2,
'express': 3,
'courier': 4,
}
tariff_val = tariff_map[tariff]
x_data = [
localize_datetime(remove_tz(x.created), 'Europe/Samara').replace(
day=1, month=1, year=1900) + datetime.timedelta(hours=4)
for x in data[list(data.keys())[0]]
]
y_data = [0 for _ in range(len(x_data))]
y_data_len = [0 for _ in range(len(x_data))]
for key in data:
for i, item in enumerate(data[key]):
try:
y = item.data['options'][tariff_val]['price']
y_data[i] += y
y_data_len[i] += 1
except Exception:
pass
y_data = [x[0] / x[1] for x in zip(y_data, y_data_len)]
return x_data, y_data
def menu_bonus(self):
datetime_now = localize_datetime(datetime.datetime.utcnow(), DEFAULT_TIME_ZONE)
datetime_last = localize_datetime(remove_tz(self.gamer.roulette_points_today), DEFAULT_TIME_ZONE)
if (datetime_now.date() - datetime_last.date()).days > 0:
self.gamer.roulette_points += 500
self.gamer.roulette_points_today = datetime_now
self.gamer.save()
return "Выдал пособие по безработице"
else:
raise PWarning("Ты уже получил бонус. Приходи завтра")
def extend_repeat_notify(self, notify):
if notify.date:
# Для постоянных уведомлений дата должа быть на завтрашний день обязательно.
# Это важно для сортировки
new_datetime = datetime.combine(
self.dt_now.date(), notify.date.time()) + timedelta(days=1)
new_datetime = localize_datetime(
remove_tz(new_datetime),
notify.user.profile.city.timezone.name)
notify.date = new_datetime
notify.save()
def get_weather(self, use_cached=True):
entity, created = Service.objects.get_or_create(
name=f'weather_{self.city.name}')
if use_cached and not created:
delta_time = (datetime.utcnow() -
remove_tz(entity.update_datetime))
if delta_time.seconds < 3600 and delta_time.days == 0:
weather_data = json.loads(entity.value)
return weather_data
weather_data = self.send_weather_request()
entity.value = json.dumps(weather_data)
entity.save()
return weather_data
def start(self):
if not self.event.sender.check_role(Role.TRUSTED) and \
len(NotifyModel.objects.filter(user=self.event.user)) >= 5:
raise PWarning("Нельзя добавлять более 5 напоминаний")
timezone = self.event.sender.city.timezone.name
date, args_count, exact_time_flag = get_time(
self.event.message.args[0], self.event.message.args[1],
self.event.sender.city.timezone)
if args_count == 2:
self.check_args(3)
if not date:
raise PWarning("Не смог распарсить дату")
date = normalize_datetime(date, timezone)
datetime_now = localize_datetime(datetime.utcnow(), "UTC")
if (date - datetime_now).seconds < 60:
raise PWarning("Нельзя добавлять напоминание на ближайшую минуту")
if not exact_time_flag and ((date - datetime_now).days < 0 or
(datetime_now - date).seconds < 0):
date = date + timedelta(days=1)
if (date - datetime_now).days < 0 or (datetime_now - date).seconds < 0:
raise PWarning("Нельзя указывать дату в прошлом")
text = self.event.message.args_str_case.split(' ',
args_count)[args_count]
if text[0] == '/':
first_space = text.find(' ')
if first_space > 0:
command = text[1:first_space]
else:
command = text[1:]
from apps.bot.commands.NotifyRepeat import NotifyRepeat
if command in self.full_names or command in NotifyRepeat(
).full_names:
text = f"/обосрать {self.event.sender.name}"
notify_datetime = localize_datetime(remove_tz(date), timezone)
notify = NotifyModel(
date=date,
text=text,
user=self.event.user,
chat=self.event.chat,
text_for_filter=notify_datetime.strftime("%d.%m.%Y %H:%M") + " " +
text)
notify.save()
notify.text_for_filter += f" ({notify.id})"
notify.save()
return f'Сохранил на дату {str(notify_datetime.strftime("%d.%m.%Y %H:%M"))}'
def menu_play(self):
with lock:
datetime_now = localize_datetime(datetime.datetime.utcnow(),
DEFAULT_TIME_ZONE)
winner_today = PetrovichGames.objects.filter(
chat=self.event.chat).first()
if winner_today:
datetime_last = localize_datetime(remove_tz(winner_today.date),
DEFAULT_TIME_ZONE)
if (datetime_now.date() - datetime_last.date()).days <= 0:
if winner_today.profile.gender and winner_today.profile.gender == '1':
winner_gender = "Петровна"
else:
winner_gender = "Петрович"
return f"{winner_gender} дня - {winner_today.profile}"
group_banned = Group.objects.get(name=Role.BANNED.name)
winner = PetrovichUser.objects.filter(
chat=self.event.chat, active=True).exclude(
profile__groups=group_banned).order_by("?").first()
if winner:
winner = winner.profile
else:
raise PWarning(
"Нет участников игры. Зарегистрируйтесь! /петрович рег")
winner_petrovich = PetrovichGames(profile=winner,
chat=self.event.chat)
winner_petrovich.save()
if winner.gender and winner.gender == '1':
winner_gender = "Наша сегодняшняя Петровна дня"
else:
winner_gender = "Наш сегодняшний Петрович дня"
first_message = random_event([
"Такс такс такс, кто тут у нас", "*барабанная дробь*",
"Вы готовы узнать победителя голодных игр?", "Ну шо, погнали",
"Вы не поверите...",
"Опять вы в игрульки свои играете да? Ну ладно"
])
return [
first_message,
f"{winner_gender} - {self.bot.get_mention(winner, str(winner))}"
]
def send_notify_message(bot, notify):
if notify.date:
notify_datetime = localize_datetime(
remove_tz(notify.date), notify.user.profile.city.timezone.name)
user_str = f"{bot.get_mention(notify.user.profile)}:" if notify.mention_sender else f"{notify.user.profile}:"
message = f"Напоминалка на {notify_datetime.strftime('%H:%M')}\n" \
f"{user_str}\n" \
f"{notify.text}"
else:
if notify.user.get_platform_enum() == Platform.TG:
message = f"Напоминалка по {get_tg_formatted_text_line(notify.crontab)}\n"
else:
message = f"Напоминалка по {notify.crontab}\n"
user_str = f"{bot.get_mention(notify.user.profile)}" if notify.mention_sender else f"{notify.user.profile}"
message += f"{user_str}\n" \
f"{notify.text}"
result_msg = {'text': message}
if notify.chat:
bot.parse_and_send_msgs(result_msg, notify.chat.chat_id)
else:
if not notify.text.startswith('/'):
bot.parse_and_send_msgs(result_msg, notify.user.user_id)
def start(self):
if not self.event.sender.check_role(Role.TRUSTED) and \
len(NotifyModel.objects.filter(user=self.event.sender)) >= 5:
raise PWarning("Нельзя добавлять более 5 напоминаний")
timezone = self.event.sender.city.timezone.name
crontab = None
date = None
try:
crontab = get_crontab(self.event.message.args)
args_split = self.event.message.args_str_case.split(' ', 5)
if len(args_split) > 5:
text = args_split[-1]
else:
text = ""
except Exception:
date = get_time(self.event.message.args[0])
if not date:
raise PWarning("Не смог распарсить дату")
date = normalize_datetime(date, timezone)
datetime_now = localize_datetime(datetime.utcnow(), "UTC")
if (date - datetime_now).seconds < 60:
raise PWarning(
"Нельзя добавлять напоминание на ближайшую минуту")
if (date - datetime_now).days < 0 or (datetime_now -
date).seconds < 0:
date = date + timedelta(days=1)
text = self.event.message.args_str_case.split(' ', 1)[1]
if text[0] == '/':
first_space = text.find(' ')
if first_space > 0:
command = text[1:first_space]
else:
command = text[1:]
from apps.bot.commands.Notify import Notify
if command in self.full_names or command in Notify().full_names:
text = f"/обосрать {self.event.sender.name}"
if date:
notify_datetime = localize_datetime(remove_tz(date), timezone)
if crontab:
notify = NotifyModel(crontab=crontab,
text=text,
user=self.event.user,
chat=self.event.chat,
repeat=True,
text_for_filter=f'{crontab} {text}')
else:
notify = NotifyModel(
date=date,
text=text,
user=self.event.user,
chat=self.event.chat,
repeat=True,
text_for_filter=f'{notify_datetime.strftime("%H:%M")} {text}')
notify.save()
notify.text_for_filter += f" ({notify.id})"
notify.save()
if date:
return f'Следующее выполнение - {str(notify_datetime.strftime("%d.%m.%Y %H:%M"))}'
else:
return 'Добавил напоминание'