bot.py

import os
import re
import random
# import openai
import twitchio
import requests
import datetime
import pytz
from twitchio import Message
from twitchio.ext import commands
from twitchio.errors import TwitchIOException
from dotenv import load_dotenv

# @todo: add a command to add a new command
# @todo: add a command to remove a command

load_dotenv(override=True)

bot = commands.Bot(
    # set up the bot
    irc_token=os.getenv('TMI_TOKEN'),
    client_id=os.getenv('CLIENT_ID'),
    nick=os.getenv('BOT_NICK'),
    token=os.environ['BOT_TOKEN'],
    prefix=os.environ['BOT_PREFIX'],
    client_secret=os.environ['TWITCH_CLIENT_SECRET'],
    initial_channels=[os.environ['CHANNEL']]
)

openapi_key = os.environ['OPENAI_KEY']
OPENWEATHER_KEY = os.environ['OPENWEATHER_KEY']

#function to generate response using gpt-3
#takes in users message as input, adds formatting, sends to gpt3 using openai api
# returns response from gpt3

# def generate_response(message):
#   prompt = f"{message.author.name} said: {message.message.content}"
#   response = openai.Completion.create(
#     engine="text-davinci-002",
#     prompt=prompt,
#     max_tokens=50,
#     n=1, 
#     stop=None,
#     temperature=0.9,
#   )
#   print(f"type of response: {type(response)}")
#   print(f"value of response: {response}")

#   choices = []
#   if response.choices is not None:
#     for choice in response.choices:
#       choices.append(choice)
  
#   if len(choices) > 0:
#     return choices[0].text.strip()
#   else:
#     return "I'm sorry, I don't know how to respond to that."
  


@bot.event() 
async def event_ready():
    print(f'{os.environ["BOT_NICK"]} || online')
    print(f'user id || {bot.user_id}')

@bot.event()  #@todo: fix this event so messages are printed
async def on_message_handler(channel: str, message: Message) -> None:
  assert message.author.name is not None
  assert message.content is not None
  message_dict = vars(message)
  for key, value in message_dict.items():
    print(f"{key}: {value}")

  if message.author.name.lower() == os.environ['BOT_NICK'].lower():
    return
  command_name = message.content.strip()
  
  print(f'@{message.author.name} executed {command_name}')
  await on_message_handler(channel, message)

@bot.command(name='hello')
async def say_hello(message):
    await message.send(f' sherbo4Catscream Hello, {message.author.name} <3!')

#@todo integrate generate_response and gpt3
# @bot.command(name='gpt3')
# async def gpt3_response(message):
#   #ignore message sent by bot itself
#   if message.author.name.lower == os.environ['BOT_NICK'].lower():
#     return  
#   #generate response from gpt3
#   response = generate_response(message.message.content)
#   #send response to chat
#   await message.send(response)

@bot.command(name='dice')
async def roll_dice(message):
  # remove characters from the message that aren't numbers
  print('***MESSGE***', message)
  print('***message.message***', message.message.content)
  
  if True: 
    sides = 6

  if str(message.message).isalpha() is False:
    #store the numbers in message in a variable
    #convert the variable to an integer
    sides_list = []
    for i in message.message.content:
      if i.isdigit():
        sides_list.append(i)
    for j in sides_list:
      j = int(j)
    print("message numbers as list ", sides_list)
    #join strings in sides_list into 1 string
    sides = ''.join(sides_list)
    #convert joined string into an integer
    print("after converting list to single string", sides)
    if len(sides_list) == 0: 
      sides = 6
    else:
      sides = int(sides)

  if sides:
    roll_result = random.randint(1, sides)
    await message.send(f'🎲 {message.author.name} rolled {roll_result} 🎲')
  else:
    await message.send(f'please use a positive number')

@bot.command(name='coin')
async def flip_coin(message):
  flip_result = random.choice(['heads', 'tails'])
  await message.send(f'🪙 {message.author.name} flipped {flip_result} 🪙')

# @todo: add a command to list all commands
@bot.command(name='sherbotbot')
async def sherbot_commands(message):
  '''list all available commands'''
  print(bot.commands.keys())
  commands = []
  commands = ['~' + command for command in bot.commands.keys()]
  command_str = ", ".join(commands)
  await message.send(f'🤖 here are all available sherbotbot commands: {command_str} 🤖')

@bot.command(name= 'sunsign')
async def sun_sign_command(message):
  message_content = str(message.message.content).lower()
  if len(message_content.split()) < 3:
    await message.send(f'please enter a valid birthday in the format ~sunsign month day')
    return
  words = message_content.split()[1:3]
  user_month = None
  user_day = None
  print('words', words, "message_content", message_content)
  # check if index 0 is integer or string to decide what is month or day
  month = words[0]
  day = words[1]
  print('month', month, 'day', day)
  user_month = month.lower().strip()
  user_day = int(day.strip(','))
  print('user_month', user_month, 'user_day', user_day)

  def get_birthday(message):
    astrology_signs= {
        'march': {'aries': range(21, 32), 'pisces': range(1, 21)},
        'april': {'taurus': range(20, 31), 'aries': range(1, 20)},
        'may': {'gemini': range(21, 32), 'taurus': range(1, 21)},
        'june': {'cancer': range(21, 31), 'gemini': range(1, 21)},
        'july': {'leo': range(23, 32), 'cancer': range(1, 23)},
        'august': {'virgo': range(23, 32), 'leo': range(1, 23)},
        'september': {'libra': range(23, 31), 'virgo': range(1, 23)},
        'october': {'scorpio': range(23, 32), 'libra': range(1, 23)},
        'november': {'sagittarius': range(22, 31), 'scorpio': range(1, 22)},
        'december': {'capricorn': range(22, 32), 'sagittarius': range(1, 22)},
        'january': {'aquarius': range(20, 32), 'capricorn': range(1, 20)},
        'february': {'pisces': range(19, 30), 'aquarius': range(1, 19)}
    }

    sign_for_month = astrology_signs[user_month]

    for sign, day_range in sign_for_month.items():
      if user_day in day_range:
        print('sign', sign)
        return sign
    raise ValueError('valid brithday pls')
  
  # if user calls sun-sign command, return their sign with get_birthday() function
  try:
    user_sign = get_birthday(message)
    today = datetime.datetime.now().strftime('%B %d').lower()
    user_date = f'{user_month} {user_day}'
    print('today', today, 'user_date', user_date)
    if user_sign:
      if user_date == today:
        await message.send(f'Happy Birthday!🎂 Your sun sign is {user_sign.capitalize()}.')
      else:
        await message.send(f'Your sun sign is {user_sign.capitalize()}.')
  except ValueError as e:
    print(str(e))


@bot.command(name="weather")
async def weather_command(message, location):
  url = f'https://api.openweathermap.org/data/2.5/weather?q={location}&appid={OPENWEATHER_KEY}&units=imperial'
  response = requests.get(url)
  data = response.json()
  await message.send("... fetching ur weather ...")
  if response.status_code == 200:
    temperature= data['main']['temp']
    feels_like = data['main']['feels_like']
    description = data['weather'][0]['description']
    city = data['name']
    temp_unit = 'F'
    if len(str(message.message.content).split()) == 3:
      temp_input = str(message.message.content).split()[-1]
      print(temp_input, len(str(message.message.content).split()))
      if 'c' in temp_input or 'C' in temp_input:
        temperature = round((temperature - 32) * 5/9)
        feels_like = round((feels_like - 32) * 5/9)
        temp_unit = 'C'
      if ('k' or 'K') in temp_input:
        temperature = round((temperature - 32) * 5/9 + 273.15)
        feels_like = round((feels_like - 32) * 5/9 + 273.15)
        temp_unit = 'K'
    if description:
      if description == 'clear sky':
        await message.send(f'🌞 {city} is {temperature}°{temp_unit} and {description}. It feels like {feels_like}°{temp_unit}. 🌞')
      if description == 'few clouds':
        await message.send(f'🌤️ {city} is {temperature}°{temp_unit} and {description}. It feels like {feels_like}°{temp_unit}. 🌤️')
      if description == 'scattered clouds':
        await message.send(f'🌤️ {city} is {temperature}°{temp_unit} and {description}. It feels like {feels_like}°{temp_unit}. 🌤️')
      if description == 'broken clouds':
        await message.send(f'☁️ {city} is {temperature}°{temp_unit} and {description}. It feels like {feels_like}°{temp_unit}. ☁️')
      if description == 'shower rain':
        await message.send(f'🌧️ {city} is {temperature}°{temp_unit} and {description}. It feels like {feels_like}°{temp_unit}. 🌧️')
      if description == 'rain':
        await message.send(f'🌧️ {city} is {temperature}°{temp_unit} and {description}. It feels like {feels_like}°{temp_unit}. 🌧️')
      if description == 'thunderstorm':
        await message.send(f'⛈️ {city} is {temperature}°{temp_unit} and {description}. It feels like {feels_like}°{temp_unit}. ⛈️')
      if description == 'snow':
        await message.send(f'❄️ {city} is {temperature}°{temp_unit} and {description}. It feels like {feels_like}°{temp_unit}. ❄️')
      if description == 'mist':
        await message.send(f'🌫️ {city} is {temperature}°{temp_unit} and {description}. It feels like {feels_like}°{temp_unit}. 🌫️')
      if description == 'overcast clouds':
        await message.send(f'☁️ {city} is {temperature}°{temp_unit} and {description}. It feels like {feels_like}°{temp_unit}. ☁️')
    else:
      await message.send(f'{city} is {temperature}°{temp_unit}. It feels like {feels_like}°{temp_unit}.')
  else:
    print('error', response.status_code)



@bot.command(name='draw')
async def draw_card(message):
  tarot_deck = [  "The Fool",  "The Magician",  "The High Priestess",  "The Empress",  "The Emperor",  "The Hierophant",  "The Lovers",  "The Chariot",  "Strength",  "The Hermit",  "Wheel of Fortune",  "Justice",  "The Hanged Man",  "Death",  "Temperance",  "The Devil",  "The Tower",  "The Star",  "The Moon",  "The Sun",  "Judgement",  "The World",  "Ace of Wands",  "Two of Wands",  "Three of Wands",  "Four of Wands",  "Five of Wands",  "Six of Wands",  "Seven of Wands",  "Eight of Wands",  "Nine of Wands",  "Ten of Wands",  "Page of Wands",  "Knight of Wands",  "Queen of Wands",  "King of Wands",  "Ace of Cups",  "Two of Cups",  "Three of Cups",  "Four of Cups",  "Five of Cups",  "Six of Cups",  "Seven of Cups",  "Eight of Cups",  "Nine of Cups",  "Ten of Cups",  "Page of Cups",  "Knight of Cups",  "Queen of Cups",  "King of Cups",  "Ace of Swords",  "Two of Swords",  "Three of Swords",  "Four of Swords",  "Five of Swords",  "Six of Swords",  "Seven of Swords",  "Eight of Swords",  "Nine of Swords",  "Ten of Swords",  "Page of Swords",  "Knight of Swords",  "Queen of Swords",  "King of Swords",  "Ace of Pentacles",  "Two of Pentacles",  "Three of Pentacles",  "Four of Pentacles",  "Five of Pentacles",  "Six of Pentacles",  "Seven of Pentacles",  "Eight of Pentacles",  "Nine of Pentacles",  "Ten of Pentacles",  "Page of Pentacles",  "Knight of Pentacles",  "Queen of Pentacles",  "King of Pentacles"]
  card = random.choice(tarot_deck)
  hand = []
  try:
    num_cards = int(message.message.content.split()[1])
    if num_cards < 1 or num_cards > 78:
      raise ValueError
  except (ValueError, IndexError):
    num_cards = 1
  for i in range(num_cards):
    card = random.choice(tarot_deck)
    hand.append(card)
  await message.send(f'{message.author.name} drew {hand}')
  
@bot.command(name='8ball')
async def eight_ball(message):
  eight_ball_answers = ['It is certain', 'It is decidedly so', 'Without a doubt', 'Yes, definitely', 'You may rely on it', 'As I see it, yes', 'Most likely', 'Outlook good', 'Yes', 'Signs point to yes', 'Reply hazy try again', 'Ask again later', 'Better not tell you now', 'Cannot predict now', 'Concentrate and ask again', 'Don\'t count on it', 'My reply is no', 'My sources say no', 'Outlook not so good', 'Very doubtful']
  answer = random.choice(eight_ball_answers)
  await message.send(f'Magic 8-ball says: {answer}')

@bot.command(name='subraid')
async def subraid(message):
  await message.send("sherbo4Figaroscreams sherbo4Catscream sherbo4Awoooo sherbo4Hype :sparkles: :lotus: sherbo4Love MercyWing1 sherbo4Pinocchio SHERBOT :rotating_light: RAID sherbo4Pinocchio MercyWing2 sherbo4Love :lotus: :sparkles: sherbo4Hype sherbo4Awoooo sherbo4Catscream sherbo4Figaroscreams")

@bot.command(name="pinocchio")
async def pinocchio_emote(message):
  await message.send("sherbo4Pinocchio sherbo4Awoooo sherbo4Pinocchio sherbo4Awoooo sherbo4Pinocchio sherbo4Awoooo sherbo4Pinocchio sherbo4Awoooo sherbo4Pinocchio sherbo4Awoooo sherbo4Pinocchio sherbo4Awoooo sherbo4Pinocchio sherbo4Awoooo sherbo4Pinocchio sherbo4Awoooo sherbo4Pinocchio sherbo4Awoooo sherbo4Pinocchio sherbo4Awoooo sherbo4Pinocchio")

@bot.command(name="awooo")
async def awooo_emote(message):
  await message.send("sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo sherbo4Awoooo")

@bot.command(name="hype")
async def hype_emote(message):
  await message.send("sherbo4Catscream ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Hype ✨ sherbo4Catscream")

@bot.command(name="love")
async def love_emote(message):
  await message.send("sherbo4Sherbot <3 KPOPlove <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 sherbo4Love <3 KPOPlove <3 sherbo4Sherbot ")

@bot.command(name="figaro")
async def figaro_emote(message):
  await message.send("sherbo4Catscream sherbo4Fig sherbo4Figaroscreams sherbo4Fig sherbo4Figaroscreams sherbo4Fig sherbo4Figaroscreams 🌸🌸🌸 sherbo4Fig sherbo4Figaroscreams sherbo4Fig sherbo4Figaroscreams sherbo4Fig sherbo4Figaroscreams sherbo4Catscream")

@bot.command(name="420")
async def four_twenty_emote(message):
  await message.send("🍃 sherbo4420blaze sherbo4Towlie 💨 sherbo4420blaze sherbo4Towlie 💨 sherbo4420blaze sherbo4Towlie 💨 sherbo4420blaze sherbo4Towlie 💨 sherbo4420blaze sherbo4Towlie 💨 sherbo4420blaze sherbo4Towlie 💨 sherbo4420blaze sherbo4Towlie 🍃")

@bot.command(name="battle")
async def battle_streamraiders_url(message):
  await message.send("!battle")


@bot.command(name="quote")
async def quote(message):
  await message.send("!quote")

@bot.command(name="ultra")
async def ultra(message):
  await message.send("!pokecatch ultraball")

@bot.event(name="")
async def pokemon_appears(message):
  if "Pokemon" in message.message.content:
    await message.send("sherbo4Pinocchio")
  if message.author.name == "PokemonCommunityGame" and "90s" in message.message.content:
    await message.send("!pokecheck")
  if '\U0001F4A9' in message.message.content:
    await message.send("~ultra")

@bot.command(name='test')
async def test(message):
  await message.send('test passed!')

bot.event(event_ready) # type: ignore
bot.event(on_message_handler) # type: ignore
bot.event(pokemon_appears) # type: ignore


#bot.py
try:
    if __name__ == "__main__":
        bot.run()
except TwitchIOException as e:
    print(e)