Added p1p1 functionality v1

Author: Tokugero

.gitignore

@@ -0,0 +1,8 @@
+__pycache__/
+pip-selfcheck.json
+bin
+include
+lib
+lib64
+pyvenv.cfg
+share

cubebot.py

@@ -0,0 +1,28 @@
+from discord.ext import commands
+import asyncio
+import os
+import logging
+import sys
+from functions import cubebot
+
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', datefmt='%m/%d/%Y %I:%M:%S %p')
+logger = logging.getLogger('cubebot.main')
+
+#EnvironmentLoaders
+botToken = os.environ['CUBEBOT']
+
+#Initializing bot
+description = '''CubeBot Commands'''
+bot = commands.Bot(command_prefix='$', description=description)
+
+@bot.event
+async def on_ready():
+    logger.info('Logged in as')
+    logger.info(bot.user.name)
+    logger.info(bot.user.id)
+    logger.info('------')
+
+
+bot.load_extension("functions.cubebot")
+
+bot.run(botToken)

functions/__init__.py

@@ -0,0 +1 @@
+__all__ = ["cubebot", "cubetutor"]

functions/cubebot.py

@@ -0,0 +1,36 @@
+from functions import cubetutor
+import logging
+import discord
+import sys
+import asyncio
+import random
+import requests
+from discord.ext import commands
+
+#shahrazad token ed98810941cf53fa9806f9aa9299c2 - This is only available internally and only executes the shahrazad test pipeline
+logger = logging.getLogger('cubebot.cubebot')
+
+logger.info("cubebot cog loaded")
+
+class Application:
+	def __init__(self, bot):
+		self.bot = bot
+
+	@commands.command(pass_context=True)
+	async def p1p1(self, ctx):
+		"""Generate a P1P1 image given a cubetutor cube ID"""
+		command = ctx.message.content.split()
+		if len(command) > 1:
+			try:
+				cubeId = int(command[1])
+				logger.info(str(ctx.message.author) + " pulled a pack")
+				await ctx.channel.send("Looking up pack...")
+				cubeId = command[1]
+				packLoader = self.bot.loop.create_task(cubetutor.CubeTutorPackChrome(cubeId, ctx))
+			except:
+				await ctx.channel.send("Please give me a real ID at least. ")
+		if len(command) is 1:
+			await ctx.channel.send("I need an ID.")
+
+def setup(bot):
+	bot.add_cog(Application(bot))

functions/cubetutor.py

@@ -0,0 +1,47 @@
+from selenium import webdriver
+from selenium.webdriver.common.keys import Keys
+import os
+import discord
+
+selenium = os.environ["HUB"]
+
+async def CubeTutorPackChrome(cubeId, ctx):
+    try:
+        cubeId = int(cubeId)
+    except Exception as e:
+        print(e)
+        return False
+    endpoint = "http://www.cubetutor.com/samplepack/"+str(cubeId)
+    driver = webdriver.Remote(
+        command_executor = selenium + "wd/hub",
+        desired_capabilities = {"browserName": "chrome", "javascriptEnabled": True}
+        )
+    try:
+        driver.get(endpoint)
+    except Exception as e:
+        print(e)
+        await tearDownClass(driver)
+        
+    try:
+        gtfo = webdriver.common.action_chains.ActionChains(driver)
+        footer = driver.find_element_by_id('footer')
+        gtfo.move_to_element(footer)
+        element = driver.find_element_by_id('main')
+        elementPng = element.screenshot_as_png
+        
+        await tearDownClass(driver, png=elementPng, ctx=ctx)
+    except Exception as e:
+        print(e)
+        await tearDownClass(driver)
+
+async def tearDownClass(driver, png=None, ctx=None): 
+    driver.quit()
+    if png:
+        packImage = discord.File(png, "crack.png")
+        await ctx.channel.send("You're pack, friend. ", file=packImage)
+    if not png:
+        await ctx.channel.send("Sorry, your pack couldn't be found. Please try again later.")
+
+if __name__ == "__main__":
+    setUpClass()
+    

requirements.txt

@@ -0,0 +1,23 @@
+aiohttp==3.4.4
+git+https://github.com/Rapptz/discord.py@rewrite#egg=discord.py[voice]
+async-timeout==3.0.1
+attrs==18.2.0
+certifi==2018.11.29
+cffi==1.11.5
+chardet==3.0.4
+docutils==0.14
+idna==2.8
+idna-ssl==1.1.0
+jmespath==0.9.3
+multidict==4.5.2
+pycparser==2.19
+PyNaCl==1.2.1
+python-dateutil==2.7.5
+python-gitlab==1.7.0
+requests==2.21.0
+six==1.12.0
+urllib3==1.24.1
+websockets==6.0
+yarl==1.1.1
+selenium==3.141.0
+