Started factoring out code to add a telegram bot

ksadov · Jul 31, 2022 · 553402b · 553402b
1 parent cbd8710
commit 553402b
Show file tree

Hide file tree

Showing 95 changed files with 164 additions and 12 deletions.
diff --git a/PRIVACY.md b/PRIVACY.md
@@ -0,0 +1,33 @@
+Privacy Policy
+Last Updated: Jul 28, 2022
+By using our Bot, you agree that you have read and agree to this policy.
+
+This is our "Privacy Policy" which sets out the policy which governs our use of information you provide in connection with the simple tarot bot. The terms "you" and "your" refer to all individuals or entities accessing this website. The terms "we," "us," "our," refer to the creators of Simple Tarot Bot and "bot" refers to our Discord bot itself.
+
+We may update this Privacy Policy from time to time. Changes in our Privacy Policy will be effective immediately. If you are a regular user of the bot, we recommend that you check this Privacy Policy on a regular basis. By using the bot, you consent to the collection, use and transfer of your information in accordance with this Privacy Policy. If you do not agree to this Privacy Policy, please do not use our bot.
+
+PRIVACY STATEMENT
+We respect the privacy of your information. We provide this explanation about our information practices as a show of our commitment to protect your privacy. This policy describes the types of information we may collect from you or that you may provide when you visit this website and our practices for collecting, using, maintaining, protecting and disclosing that information.
+
+Please read this policy carefully to understand our policies and practices regarding your information and how we treat it. If you do not agree with our policies and practices, your choice is not to use our bot. By accessing or using this bot, you agree to this privacy policy. This policy may change from time to time (see Changes to Our Privacy Policy). Your continued use of this bot after we make changes is deemed to be acceptance of those changes, so please check the policy periodically for updates.
+
+THE INFORMATION WE COLLECT AND HOW WE COLLECT IT
+When you use the bot and save settings, we collect and store your discord user id, as well as the preferences you specify.
+
+THIRD-PARTY WEBSITES
+Messages distributed by our bot may contain links to third-party websites. These linked websites are not under our control, and we are not responsible for the privacy practices or the contents of any such linked website or any link contained in any linked website. We provide such links only as a convenience, and the inclusion of a link on the website does not imply endorsement of the linked website by Simple Tarot Bot. If you provide any personal data through any such third-party website, your transaction will occur on the third party’s website (not this website) and the personal data you provide will be collected by and controlled by the privacy policy of that third party. We recommend that you familiarize yourself with the privacy policies and practices of any third parties. PLEASE NOTE THAT THIS PRIVACY POLICY DOES NOT ADDRESS THE PRIVACY OR INFORMATION PRACTICES OF ANY THIRD PARTIES.
+
+THE WAY WE USE INFORMATION
+This Privacy Statement governs use of the information that you provide to us through the bot. It does not govern the manner in which we may use information obtained from other sources (such as public records) or from you by means other than this bot. We and other entities that we are involved in, or provide support for, the operation of this bot or other purposes use information that we collect about you or that you provide to us, including any personal information:
+
+To provide you with the services of the bot using your specified preferences.
+In any other way we may describe when you provide the information.
+For any other purpose with your consent.
+We may also use your information if provided to us to contact you about our own and third-party services and products that may be of interest to you. If you do not want us to use your information in this way, you can opt out of receiving this information by so indicating in your registration/preferences file or by notifying us.
+
+CHANGES TO OUR PRIVACY POLICY
+It is our policy to post any changes we make to our privacy policy on this page. If we make material changes to how we treat our users’ personal information, we will notify you by e-mail to the primary e-mail address specified in your registration information if any has been established and/or through a notice on the website home page. The date the privacy policy was last revised is identified at the top of the page. If information has been collected, you are responsible for ensuring we have an up-to- date active and deliverable e-mail address for you, and for periodically visiting our website and this privacy policy to check for any changes.
+
+
+CONTACT INFORMATION
+To ask questions or comment about this Privacy Policy and our privacy practices, please contact us via Discord.
diff --git a/common/__init__.py b/common/__init__.py
diff --git a/layouts.py → common/layouts.py b/layouts.py → common/layouts.py
diff --git a/tarot.py → common/tarot.py b/tarot.py → common/tarot.py
@@ -33,6 +33,7 @@ def __new__(cls, shortname, label, longname, global_d = True):
 
     DEFAULT = ("default", "Default", "Default cards")
     SWISS = ("swiss", "Swiss", "IJJ Swiss cards")
+    PLANET_SCUM = ("planetscum", "Planet Scum", "Planet Scum Custom Cards", False)
 
 @unique
 class MajorMinor(Enum):
@@ -358,7 +359,7 @@ def makeImgList (cards: List[Card], deck: Decks):
     """Returns a list of Images corresponding to cards."""
     imgarray = []
     for c in cards:
-        newcard = Image.open(path.join("decks",deck.shortname,c.code + ".jpg")).convert("RGBA")
+        newcard = Image.open(path.join("..","decks",deck.shortname,c.code + ".jpg")).convert("RGBA")
         if not c.up:
             newcardrev = newcard.rotate(180, expand = 1)
             imgarray.append(newcardrev)

diff --git a/decks/planetscum/0M.jpg b/decks/planetscum/0M.jpg
diff --git a/decks/planetscum/10C.jpg b/decks/planetscum/10C.jpg
diff --git a/decks/planetscum/10M.jpg b/decks/planetscum/10M.jpg
diff --git a/decks/planetscum/10P.jpg b/decks/planetscum/10P.jpg
diff --git a/decks/planetscum/10S.jpg b/decks/planetscum/10S.jpg
diff --git a/decks/planetscum/10W.jpg b/decks/planetscum/10W.jpg
diff --git a/decks/planetscum/11M.jpg b/decks/planetscum/11M.jpg
diff --git a/decks/planetscum/12M.jpg b/decks/planetscum/12M.jpg
diff --git a/decks/planetscum/13M.jpg b/decks/planetscum/13M.jpg
diff --git a/decks/planetscum/14M.jpg b/decks/planetscum/14M.jpg
diff --git a/decks/planetscum/15M.jpg b/decks/planetscum/15M.jpg
diff --git a/decks/planetscum/16M.jpg b/decks/planetscum/16M.jpg
diff --git a/decks/planetscum/17M.jpg b/decks/planetscum/17M.jpg
diff --git a/decks/planetscum/18M.jpg b/decks/planetscum/18M.jpg
diff --git a/decks/planetscum/19M.jpg b/decks/planetscum/19M.jpg
diff --git a/decks/planetscum/1M.jpg b/decks/planetscum/1M.jpg
diff --git a/decks/planetscum/20M.jpg b/decks/planetscum/20M.jpg
diff --git a/decks/planetscum/21M.jpg b/decks/planetscum/21M.jpg
diff --git a/decks/planetscum/2C.jpg b/decks/planetscum/2C.jpg
diff --git a/decks/planetscum/2M.jpg b/decks/planetscum/2M.jpg
diff --git a/decks/planetscum/2P.jpg b/decks/planetscum/2P.jpg
diff --git a/decks/planetscum/2S.jpg b/decks/planetscum/2S.jpg
diff --git a/decks/planetscum/2W.jpg b/decks/planetscum/2W.jpg
diff --git a/decks/planetscum/3C.jpg b/decks/planetscum/3C.jpg
diff --git a/decks/planetscum/3M.jpg b/decks/planetscum/3M.jpg
diff --git a/decks/planetscum/3P.jpg b/decks/planetscum/3P.jpg
diff --git a/decks/planetscum/3S.jpg b/decks/planetscum/3S.jpg
diff --git a/decks/planetscum/3W.jpg b/decks/planetscum/3W.jpg
diff --git a/decks/planetscum/4C.jpg b/decks/planetscum/4C.jpg
diff --git a/decks/planetscum/4M.jpg b/decks/planetscum/4M.jpg
diff --git a/decks/planetscum/4P.jpg b/decks/planetscum/4P.jpg
diff --git a/decks/planetscum/4S.jpg b/decks/planetscum/4S.jpg
diff --git a/decks/planetscum/4W.jpg b/decks/planetscum/4W.jpg
diff --git a/decks/planetscum/5C.jpg b/decks/planetscum/5C.jpg
diff --git a/decks/planetscum/5M.jpg b/decks/planetscum/5M.jpg
diff --git a/decks/planetscum/5P.jpg b/decks/planetscum/5P.jpg
diff --git a/decks/planetscum/5S.jpg b/decks/planetscum/5S.jpg
diff --git a/decks/planetscum/5W.jpg b/decks/planetscum/5W.jpg
diff --git a/decks/planetscum/6C.jpg b/decks/planetscum/6C.jpg
diff --git a/decks/planetscum/6M.jpg b/decks/planetscum/6M.jpg
diff --git a/decks/planetscum/6P.jpg b/decks/planetscum/6P.jpg
diff --git a/decks/planetscum/6S.jpg b/decks/planetscum/6S.jpg
diff --git a/decks/planetscum/6W.jpg b/decks/planetscum/6W.jpg
diff --git a/decks/planetscum/7C.jpg b/decks/planetscum/7C.jpg
diff --git a/decks/planetscum/7M.jpg b/decks/planetscum/7M.jpg
diff --git a/decks/planetscum/7P.jpg b/decks/planetscum/7P.jpg
diff --git a/decks/planetscum/7S.jpg b/decks/planetscum/7S.jpg
diff --git a/decks/planetscum/7W.jpg b/decks/planetscum/7W.jpg
diff --git a/decks/planetscum/8C.jpg b/decks/planetscum/8C.jpg
diff --git a/decks/planetscum/8M.jpg b/decks/planetscum/8M.jpg
diff --git a/decks/planetscum/8P.jpg b/decks/planetscum/8P.jpg
diff --git a/decks/planetscum/8S.jpg b/decks/planetscum/8S.jpg
diff --git a/decks/planetscum/8W.jpg b/decks/planetscum/8W.jpg
diff --git a/decks/planetscum/9C.jpg b/decks/planetscum/9C.jpg
diff --git a/decks/planetscum/9M.jpg b/decks/planetscum/9M.jpg
diff --git a/decks/planetscum/9P.jpg b/decks/planetscum/9P.jpg
diff --git a/decks/planetscum/9S.jpg b/decks/planetscum/9S.jpg
diff --git a/decks/planetscum/9W.jpg b/decks/planetscum/9W.jpg
diff --git a/decks/planetscum/AC.jpg b/decks/planetscum/AC.jpg
diff --git a/decks/planetscum/AP.jpg b/decks/planetscum/AP.jpg
diff --git a/decks/planetscum/AS.jpg b/decks/planetscum/AS.jpg
diff --git a/decks/planetscum/AW.jpg b/decks/planetscum/AW.jpg
diff --git a/decks/planetscum/KC.jpg b/decks/planetscum/KC.jpg
diff --git a/decks/planetscum/KNC.jpg b/decks/planetscum/KNC.jpg
diff --git a/decks/planetscum/KNP.jpg b/decks/planetscum/KNP.jpg
diff --git a/decks/planetscum/KNS.jpg b/decks/planetscum/KNS.jpg
diff --git a/decks/planetscum/KNW.jpg b/decks/planetscum/KNW.jpg
diff --git a/decks/planetscum/KP.jpg b/decks/planetscum/KP.jpg
diff --git a/decks/planetscum/KS.jpg b/decks/planetscum/KS.jpg
diff --git a/decks/planetscum/KW.jpg b/decks/planetscum/KW.jpg
diff --git a/decks/planetscum/PC.jpg b/decks/planetscum/PC.jpg
diff --git a/decks/planetscum/PP.jpg b/decks/planetscum/PP.jpg
diff --git a/decks/planetscum/PS.jpg b/decks/planetscum/PS.jpg
diff --git a/decks/planetscum/PW.jpg b/decks/planetscum/PW.jpg
diff --git a/decks/planetscum/QC.jpg b/decks/planetscum/QC.jpg
diff --git a/decks/planetscum/QP.jpg b/decks/planetscum/QP.jpg
diff --git a/decks/planetscum/QS.jpg b/decks/planetscum/QS.jpg
diff --git a/decks/planetscum/QW.jpg b/decks/planetscum/QW.jpg
diff --git a/discordbot/__init__.py b/discordbot/__init__.py
diff --git a/changeGuildSettings.py → discordbot/changeGuildSettings.py b/changeGuildSettings.py → discordbot/changeGuildSettings.py
@@ -1,5 +1,5 @@
 import shelve
-from tarot import Decks
+from ..common.tarot import Decks
 
 # add non-standard deck to the guilds
 def addCustomDeck(guildid, deck):
@@ -11,4 +11,7 @@ def addCustomDeck(guildid, deck):
 
 #put custom deck into 605900319922061342
 if __name__ == "__main__":
-    # addCustomDeck(605900319922061342, )
+    addCustomDeck("605900319922061342", Decks.PLANET_SCUM)
+    # test servers. remove these later
+    addCustomDeck("357633267861553162", Decks.PLANET_SCUM)
+    addCustomDeck("410850945229127692", Decks.PLANET_SCUM)
diff --git a/components.py → discordbot/components.py b/components.py → discordbot/components.py
@@ -1,5 +1,5 @@
 import discord
-from tarot import ReadingType, Decks, MajorMinor
+from ..common.tarot import ReadingType, Decks, MajorMinor
 from handler import handle, READING_DEFAULTS
 import shelve
 from itertools import chain

diff --git a/discordslash.py → discordbot/discordslash.py b/discordslash.py → discordbot/discordslash.py
@@ -1,6 +1,6 @@
 import os
-import tarot
-from tarot import ReadingType, Decks
+from ..common import tarot
+from ..common.tarot import ReadingType, Decks
 import discord
 from dotenv import load_dotenv
 from io import BytesIO

diff --git a/handler.py → discordbot/handler.py b/handler.py → discordbot/handler.py
@@ -1,7 +1,7 @@
 import discord
 from io import BytesIO
-import tarot
-from tarot import ReadingType, Decks, MajorMinor
+from ..common import tarot
+from ..common.tarot import ReadingType, Decks, MajorMinor
 import shelve
 
 color = discord.Colour(0x6b1bf8)

diff --git a/main.py → discordbot/main.py b/main.py → discordbot/main.py
@@ -1,6 +1,6 @@
 # attempting to switch to pycord because interactions has annoyed me
 import os
-from tarot import ReadingType
+from ..common.tarot import ReadingType
 from dotenv import load_dotenv
 import discord
 from components import ReadingSelectorView, AboutView, SettingsView
@@ -60,4 +60,8 @@ async def _do_reading(ctx):
     addCommand(t)
     help_message += "/{}: {}\n".format(t.id, t.description)
 
-bot.run(token)
+def main():
+    bot.run(token)
+
+if __name__ == '__main__':
+    main()
diff --git a/main_interactions.py → discordbot/main_interactions.py b/main_interactions.py → discordbot/main_interactions.py
@@ -1,8 +1,8 @@
 import interactions
 from interactions import Embed, EmbedField, EmbedImageStruct, Attachment
 import os
-import tarot
-from tarot import ReadingType, Decks
+from ..common import tarot
+from ..common.tarot import ReadingType, Decks
 from dotenv import load_dotenv
 from io import BytesIO
 from components import createComponents, default_components, about_components, settings_components, start_reading_components

diff --git a/discordbot/numServers.py b/discordbot/numServers.py
@@ -0,0 +1,13 @@
+from dotenv import load_dotenv
+import discord
+import os
+
+load_dotenv()
+bot = discord.Bot()
+
+@bot.event
+async def on_ready():
+    print("bot is running")
+    print(len(bot.guilds))
+
+bot.run(os.getenv('DISCORD_TOKEN'))
diff --git a/requirements.txt b/requirements.txt
@@ -1,15 +1,22 @@
 aiodns==3.0.0
 aiohttp==3.8.1
 aiosignal==1.2.0
+anyio==3.6.1
+APScheduler==3.9.1
 async-timeout==4.0.2
 attrs==21.4.0
 brotlipy==0.7.0
+cachetools==5.2.0
 cchardet==2.1.7
+certifi==2022.6.15
 cffi==1.15.0
 chardet==4.0.0
 charset-normalizer==2.0.12
 colorama==0.4.4
 frozenlist==1.3.0
+h11==0.12.0
+httpcore==0.15.0
+httpx==0.23.0
 idna==3.3
 multidict==6.0.2
 orjson==3.6.8
@@ -18,5 +25,14 @@ py-cord==2.0.0b7
 pycares==4.1.2
 pycparser==2.21
 python-dotenv==0.20.0
+python-telegram-bot==20.0a1
+pytz==2022.1
+pytz-deprecation-shim==0.1.0.post0
+rfc3986==1.5.0
+six==1.16.0
+sniffio==1.2.0
+tornado==6.1
 typing_extensions==4.2.0
+tzdata==2022.1
+tzlocal==4.2
 yarl==1.7.2
diff --git a/runDiscord.py b/runDiscord.py
@@ -0,0 +1,4 @@
+from discordbot import main
+
+if __name__ == '__main__':
+    main.main()
diff --git a/runTelegram.py b/runTelegram.py
@@ -0,0 +1,4 @@
+from telegrambot import main
+
+if __name__ == '__main__':
+    main.main()
diff --git a/telegrambot/__init__.py b/telegrambot/__init__.py
diff --git a/telegrambot/main.py b/telegrambot/main.py
@@ -0,0 +1,74 @@
+import os
+from uuid import uuid4
+from telegram import Update, InlineQueryResultArticle, InlineQueryResultPhoto, InputTextMessageContent
+from telegram.ext import ApplicationBuilder, ContextTypes, CommandHandler, filters, MessageHandler, InlineQueryHandler
+from dotenv import load_dotenv
+
+load_dotenv()
+
+startText = "I'm TarotBot! You can get a tarot reading with /tarot, or update your settings with /settings. Get help with /help"
+
+# When a user starts interaction with a bot
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    await context.bot.send_message(
+        chat_id=update.effective_chat.id,
+        text=startText
+    )
+
+# For when an unknown command is sent
+async def unknown(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    await context.bot.send_message(chat_id=update.effective_chat.id, text="Sorry, I didn't understand that command.")
+
+# Help command
+async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    await context.bot.send_message(
+        chat_id=update.effective_chat.id,
+        text=startText
+    )
+
+# TODO: edit settings
+async def change_settings(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    pass
+
+
+# TODO: conduct a reading
+# TODO: inline query
+async def inline_query(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    """Handle the inline query. This is run when you type: @botusername <query>"""
+    # query = update.inline_query.query
+
+    # if query == "":
+    #     return
+
+    results = [
+        InlineQueryResultPhoto(
+            id=str(uuid4()),
+            photo_url=???,
+            title=???
+            description=???,
+            caption=???
+        )
+    ]
+
+    await update.inline_query.answer(
+        results,
+        switch_pm_text="Change Tarot Settings"
+    )
+
+
+def main():
+    application = ApplicationBuilder().token(os.getenv('TELEGRAM_API_TOKEN')).build()
+
+    application.add_handler(CommandHandler('start', start))
+    application.add_handler(CommandHandler('settings', change_settings))
+    application.add_handler(CommandHandler('help', help_command))
+    application.add_handler(InlineQueryHandler(inline_query))
+
+    # Should be the last handler
+    application.add_handler(MessageHandler(filters.COMMAND, unknown))
+
+
+    application.run_polling()
+
+if __name__ == '__main__':
+    main()