From f2752967d35d17d13694b39fc6b543da497bd4b4 Mon Sep 17 00:00:00 2001
From: Sakimori <sakimorihalley@gmail.com>
Date: Sat, 8 Jan 2022 00:00:22 -0500
Subject: [PATCH] Got twitter bot working, have player attribute responses

---
 .gitignore     |   5 ++-
 SimHoc.py      |  22 ++++++++++
 SimHoc.pyproj  |  17 ++++++++
 attributes.py  | 115 +++++++++++++++++++++++++++++++++++++++++++++++++
 db.py          |  19 ++++++++
 player.py      |  44 +++++++++++++++++++
 twitHandler.py |  45 +++++++++++++++++++
 7 files changed, 266 insertions(+), 1 deletion(-)
 create mode 100644 attributes.py
 create mode 100644 db.py
 create mode 100644 player.py
 create mode 100644 twitHandler.py

diff --git a/.gitignore b/.gitignore
index 2813ce2..6d23af2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -363,4 +363,7 @@ MigrationBackup/
 FodyWeavers.xsd
 
 # Venv files
-Hockey/
\ No newline at end of file
+Hockey/
+
+# Data directiory
+Data/
\ No newline at end of file
diff --git a/SimHoc.py b/SimHoc.py
index 8b13789..d8b52ac 100644
--- a/SimHoc.py
+++ b/SimHoc.py
@@ -1 +1,23 @@
+import os, pygame, player, tweepy, twitHandler, time
 
+if __name__ == "__main__":
+    #for name in ["Vivi", "Artemis", "Laika", "Sharks", "Dragons", "Melua", "Sabriina", "Jorts (Buttered)", "Jorts (Unbuttered)"]:
+    #    plyr = player.Player(name)
+    #    print(f"{name}:")
+    #    for atr in plyr.attributes:
+    #        print(atr)
+    #    print("----------")
+
+    twitter = twitHandler.TwitHandler()
+    if os.path.exists(os.path.join("Data", "lastID.twt")):
+        with open(os.path.join("Data", "lastID.twt")) as idFile:
+            lastID = idFile.readline().strip()
+    else:
+        lastID = 0
+
+    while True:
+        twitter.scanForMention(lastID)
+        time.sleep(30)
+        with open(os.path.join("Data", "lastID.twt")) as idFile:
+            lastID = idFile.readline().strip()
+    #twitter.sendTextTweet(player.Player("Amogus").twitterString())
\ No newline at end of file
diff --git a/SimHoc.pyproj b/SimHoc.pyproj
index c18bf7b..e3fa69d 100644
--- a/SimHoc.pyproj
+++ b/SimHoc.pyproj
@@ -22,6 +22,16 @@
     <EnableUnmanagedDebugging>false</EnableUnmanagedDebugging>
   </PropertyGroup>
   <ItemGroup>
+    <Compile Include="attributes.py" />
+    <Compile Include="twitHandler.py">
+      <SubType>Code</SubType>
+    </Compile>
+    <Compile Include="db.py">
+      <SubType>Code</SubType>
+    </Compile>
+    <Compile Include="player.py">
+      <SubType>Code</SubType>
+    </Compile>
     <Compile Include="SimHoc.py" />
   </ItemGroup>
   <ItemGroup>
@@ -35,6 +45,13 @@
       <Architecture>X86</Architecture>
     </Interpreter>
   </ItemGroup>
+  <ItemGroup>
+    <Folder Include="Data\" />
+  </ItemGroup>
+  <ItemGroup>
+    <Content Include="Data\s options.txt" />
+    <Content Include="Data\Twitter.keys" />
+  </ItemGroup>
   <Import Project="$(MSBuildExtensionsPath32)\Microsoft\VisualStudio\v$(VisualStudioVersion)\Python Tools\Microsoft.PythonTools.targets" />
   <!-- Uncomment the CoreCompile target to enable the Build command in
        Visual Studio and specify your pre- and post-build commands in
diff --git a/attributes.py b/attributes.py
new file mode 100644
index 0000000..f5f3a37
--- /dev/null
+++ b/attributes.py
@@ -0,0 +1,115 @@
+import math, random
+
+versionNumber = 1
+
+#Class and function definitions
+class Attribute:
+    """Contains a name string and a value float."""
+    def __init__(self, name:str, value:float):
+        self.name = name
+        self.value = value
+
+    def newValue(self, new):
+        self.value = new
+
+    def twitterFormat(self):
+        frac, whole = math.modf(self.value/20)
+        valueString = ""
+        valueString += "🟩"*int(whole) #green square
+        if frac > 0.1 and frac < 0.5:
+            valueString += "🟧" #orange square
+        elif frac >= 0.5 and frac < 0.9:
+            valueString += "🟨" #yellow square
+        elif frac > 0.9:
+            valueString += "🟩" #green square
+        while len(valueString) < 5:
+            valueString += "🟥" #red square
+        return f"{valueString} - {self.name}"
+
+    def __str__(self):
+        return f"{self.name}: {int(self.value)}"
+
+    #Comparison helpers
+    def __eq__(self, other): #can compare attributes by name, or find attribute by value
+        if isinstance(other, Attribute):
+            return self.name == other.name
+        elif isinstance(other, int):
+            return self.value == other
+        else:
+            return False
+
+    def __ne__(self, other):
+        return not self == other
+
+    #less than/greater than compare attributes by value, or compare value to int
+    def __lt__(self, value):
+        if isinstance(other, Attribute):
+            return self.value < other.value
+        elif isinstance(other, int):
+            return self.value < other
+        else:
+            return False
+
+    def __le__(self, value):
+        if isinstance(other, Attribute):
+            return self.value <= other.value
+        elif isinstance(other, int):
+            return self.value <= other
+        else:
+            return False
+
+    def __gt__(self, other):
+        if isinstance(other, Attribute):
+            return self.value > other.value
+        elif isinstance(other, int):
+            return self.value > other
+        else:
+            return False
+
+    def __ge__(self, other):
+        if isinstance(other, Attribute):
+            return self.value >= other.value
+        elif isinstance(other, int):
+            return self.value >= other
+        else:
+            return False
+        
+        
+
+def normalDis(generator:random.Random, mean:float, stdDev:float, min:float=-math.inf, max:float=math.inf) -> float:
+    """Generates random number from normal distribution with given mean and standard deviation. Optionally takes min and max values."""
+    num = generator.gauss(mean, stdDev)
+    while min > num or num > max:
+        num = generator.gauss(mean, stdDev)
+    return num
+
+#Attributes to generate
+#["sample name", mean, stdDev, min(optional), max(optional)]
+masterList = [
+    ["Speed", 60, 25, 0, 100], #sprint speed (Forward)
+    ["Agility", 50, 35, 0, 100], #changing directions/goaltending slide (Goalie)
+    ["Reflexes", 50, 35, 0, 100], #deflections/goaltending blocks/intercepting passes (Goalie)
+    ["Strength", 40, 60, 0, 100], #power of slapshot/body checking (Defense)
+    ["Dexterity", 50, 25, 10, 100], #power of wrist shot/dekes (Forward)
+    ["Shot Accuracy", 40, 25, 10, 100], #accuracy on shots (Secondary/All Skaters)
+    ["Pass Accuracy", 60, 35, 0, 100], #does what you think it does (Defense)
+    ["Wisdom", 40, 50, 0, 100], #takes better shots/makes safer passes (Secondary/All)
+    ["Stickhandling", 50, 35, 0, 100], #dekes/checking (Secondary/All)
+    ["Discipline", 75, 60, 0, 100], #Higher means less penalties
+    ["Intelligence", 50, 50, 0, 100], #Skill at positioning (as skater or goalie) (Secondary/All)
+    ["Constitution", 60, 20, 0, 100] #Stamina/injury resilience
+    ]
+
+noPrint = [
+    "Discipline",
+    "Constitution"
+    ]
+    
+def attributesFromName(name:str):
+    """Returns a List of Attribute objects, seeded by the name parameter."""
+    generator = random.Random() #individual random instance to avoid conflicts
+    generator.seed(name)
+    atrs = [versionNumber] #in case of stat changes, record which number to use
+    for template in masterList:
+        atrs.append(Attribute(template[0], normalDis(generator, *template[1:])))
+    return atrs
\ No newline at end of file
diff --git a/db.py b/db.py
new file mode 100644
index 0000000..fc96582
--- /dev/null
+++ b/db.py
@@ -0,0 +1,19 @@
+import os
+import sqlite3 as sql
+
+dataDir = "Data"
+dbName = "NSHL.db" 
+
+def connect():
+    #create connection, create db if doesn't exist
+    conn = None
+    try:
+        conn = sql.connect(os.path.join(data_dir, dbName))
+
+        # enable write-ahead log for performance and resilience
+        conn.execute('pragma journal_mode=wal')
+
+        return conn
+    except:
+        print("oops, db connection no work")
+        return conn
diff --git a/player.py b/player.py
new file mode 100644
index 0000000..6385ec4
--- /dev/null
+++ b/player.py
@@ -0,0 +1,44 @@
+import attributes
+
+class CreationError(Exception):
+    pass
+
+class Player(object):
+    """A hockey player with attributes and various functions."""
+
+    def __init__(self, name:str):
+        if len(name) > 30:
+            raise CreationError("Player name too long.")
+        self.name = name
+        self.attributes = self.loadAttributes()
+
+    def loadAttributes(self):
+        """Generates attributes based on name, or loads from database if present."""
+        rawAtrs = attributes.attributesFromName(self.name)
+        self.attributesVersion = rawAtrs[0]
+        return rawAtrs[1:]
+
+    def twitterString(self): 
+        """Generates a twitter-formatted string representing the player."""
+        send = f"{self.name}:\n"
+        for attr in self.attributes:
+            if attr.name not in attributes.noPrint:
+                send += attr.twitterFormat()
+                send += "\n"
+        return send
+
+    def __eq__(self, value):
+        if isinstance(value, Player):
+            return self.name == value.name
+        elif isinstance(value, str):
+            return self.name == value
+        else:
+            return False
+
+
+class Skater(Player):
+    """A hockey player that is not a goalie."""
+
+
+class Goalie(Player):
+    """A hockey player that *is* a goalie."""
\ No newline at end of file
diff --git a/twitHandler.py b/twitHandler.py
new file mode 100644
index 0000000..32f9b8d
--- /dev/null
+++ b/twitHandler.py
@@ -0,0 +1,45 @@
+import tweepy, os, json, player
+
+dataDir = "Data"
+
+class TwitHandler(object):
+    """Twitter connection handler class"""
+    api = tweepy.Client()
+
+    def __init__(self):
+        path = os.path.join(dataDir, "Twitter.keys")
+        if os.path.exists(path):
+            with open(path) as keysFile:
+                bearerToken, consumerKey, consumerSecret, accessKey, accessSecret = [line.strip() for line in keysFile.readlines()]
+        else:
+            raise FileNotFoundError
+        self.api = tweepy.Client(bearerToken, consumerKey, consumerSecret, accessKey, accessSecret)
+
+    def scanForMention(self, lastRepliedID):
+        mentions = self.api.get_users_mentions(1479541275862908928, since_id=lastRepliedID, max_results=20)
+        if mentions.data is None or len(mentions.data) == 0:
+            return
+
+        for mention in reversed(mentions.data): #do oldest first
+            lastID = mention.id
+
+            if "rate " in mention.text.lower():
+                try:
+                    name = mention.text.split("rate ",1)[1]
+                    self.sendTextReply(player.Player(name).twitterString(), mention)
+                except:
+                    print("Tweet already replied to.")
+
+        with open(os.path.join(dataDir, "lastID.twt"), 'w') as file:
+            file.write(str(lastID+1))
+
+
+
+    def sendTextTweet(self, text:str):
+        self.api.create_tweet(text=text)
+
+    def sendTextReply(self, text:str, prevTweet):
+        self.api.create_tweet(in_reply_to_tweet_id=prevTweet.id, text=text)
+
+    def changeBio(self, newText):
+        pass #awaiting API v1 permission
\ No newline at end of file