async def send_code(ctx, group_name, users: commands.Greedy[User], is_authorized_or_owner=is_authorized_or_owner): """Sends a promo code from a code group to each of the mentioned users.""" logging.info( "Tentando enviar um código do grupo %s para o(s) usuário(s) %s", group_name, ', '.join([f'{user.name}({user.id})' for user in users])) group = PromoCodeGroup.get_or_none( (PromoCodeGroup.guild_id == ctx.guild.id) & (PromoCodeGroup.name == group_name)) if group is None: await ctx.send("Grupo {} não existe".format(group_name)) return messages_author = [] messages_channel = [] with db.atomic() as transaction: for user in users: if not await is_authorized_or_owner(ctx): used_codes = PromoCode.select().where( (PromoCode.group == group) & (PromoCode.sent_to_id == user.id)) if used_codes.count() > 0: messages_channel.append( "Usuário {0} já resgatou código do grupo {1}".format( user.name, group_name)) continue promo_code = PromoCode.select().where( (PromoCode.group == group) & (PromoCode.sent_to_id == None) # noqa E501 pylint: disable=singleton-comparison ).first() if promo_code is None: messages_channel.append( "Grupo {} não possui mais códigos disponíveis".format( group_name)) break promo_code.sent_to_name = user.name promo_code.sent_to_id = user.id promo_code.sent_at = datetime.now(timezone.utc) promo_code.save() await user.send("Olá! Você ganhou um código: {}".format( promo_code.code)) # incluir essa linha nos testes! messages_author.append( "Código {} enviado para o usuário {}".format( promo_code.code, user.name)) messages_channel.append( "Enviado código do grupo {} para o usuário {}".format( group_name, user.name)) await ctx.send("\n".join(messages_channel)) await ctx.author.send("\n".join(messages_author)) transaction.commit()
async def remove_code(ctx, group_name, code): """Removes a code from a code group.""" logging.info("Tentando remover o código %s do grupo %s", code, group_name) group = PromoCodeGroup.get_or_none( (PromoCodeGroup.guild_id == ctx.guild.id) & (PromoCodeGroup.name == group_name)) if group is None: await ctx.send("Código {0} não encontrado no grupo {1}".format( code, group_name)) return query = PromoCode.delete().where((PromoCode.code == code) & (PromoCode.group == group)) rows_removed = query.execute() if rows_removed > 0: await ctx.send("Código {0} excluído do grupo {1}".format( code, group_name)) else: await ctx.send("Código {0} não encontrado no grupo {1}".format( code, group_name))
async def list_code(ctx, group_name): """Lists all codes inside a code group.""" logging.info("Tentando listar os códigos do grupo %s", group_name) group = PromoCodeGroup.get_or_none( (PromoCodeGroup.guild_id == ctx.guild.id) & (PromoCodeGroup.name == group_name)) if group is None: await ctx.send("Grupo {} não existe".format(group_name)) return codes = PromoCode.select().where((PromoCode.group == group)) if codes.count() == 0: # pylint: disable=no-value-for-parameter await ctx.send("Grupo {} não possui códigos".format(group_name)) return output = "Códigos para o grupo {}: ".format(group_name) for code in codes: output += "\n- {}".format(code.code) sent_at = sqlite_datetime_hack(code.sent_at) if code.sent_to_id: output += " enviado para o usuário {0} em {1}".format( code.sent_to_name, sent_at.astimezone(LOCAL_TIMEZONE).strftime(DATETIME_FORMAT)) await send_long_message_array(ctx.author.send, output)
async def add_code_bulk(ctx, group_name, *, code_bulk): """Adds a lot of codes to the same group. Codes are separated by anything that is not: - a letter - a number - a dash (-)""" logging.info("Tentando adicionar códigos em massa ao grupo %s: %s", group_name, code_bulk) group = PromoCodeGroup.get_or_none( (PromoCodeGroup.guild_id == ctx.guild.id) & (PromoCodeGroup.name == group_name)) if group is None: await ctx.send( "Grupo de códigos promocionais não encontrado: {}".format( group_name)) return codes = parse_codes_in_bulk(code_bulk) insert_bulk_data = [{'group': group, 'code': code} for code in codes] with db.atomic() as transaction: PromoCode.insert_many(insert_bulk_data).execute() # noqa E501 pylint: disable=no-value-for-parameter transaction.commit() await ctx.send("Códigos adicionados ao grupo {}".format(group_name))