first commit

README.md

@@ -0,0 +1 @@
+"# TFC" 

TFCmaps.py

@@ -0,0 +1,143 @@
+# bot.py
+from __future__ import print_function
+
+import os
+import random
+
+from discord.ext import commands
+from dotenv import load_dotenv
+
+from google.auth.transport.requests import Request
+from google.oauth2.credentials import Credentials
+from google_auth_oauthlib.flow import InstalledAppFlow
+from googleapiclient.discovery import build
+from googleapiclient.errors import HttpError
+
+# If modifying these scopes, delete the file token.json.
+SCOPES = ['https://www.googleapis.com/auth/spreadsheets.readonly']
+
+# The ID and range of Neon's sheet of maps + team sizes.
+SAMPLE_SPREADSHEET_ID = '1sXUDGC9XrRqKNbjq-qZeMNIgQDeRXdimEY1PZmXt6cY'
+SAMPLE_RANGE_NAME = 'Full List!A3:D'
+
+def pullmaps(players='2'):
+    pulledmaps = [] # i don't think this is necessary
+    creds = None
+    # The file token.json stores the user's access and refresh tokens, and is
+    # created automatically when the authorization flow completes for the first
+    # time.
+    if os.path.exists('token.json'):
+        creds = Credentials.from_authorized_user_file('token.json', SCOPES)
+    # If there are no (valid) credentials available, let the user log in.
+    if not creds or not creds.valid:
+        if creds and creds.expired and creds.refresh_token:
+            creds.refresh(Request())
+        else:
+            flow = InstalledAppFlow.from_client_secrets_file(
+                'credentials.json', SCOPES)
+            creds = flow.run_local_server(port=0)
+        # Save the credentials for the next run
+        with open('token.json', 'w') as token:
+            token.write(creds.to_json())
+
+    #check playercount so we can get the right maps
+    match players:
+        case '3':
+            playercount = '3v3'
+        case '4':
+            playercount = '4v4'
+        case '5':
+            playercount = '5v5'
+        case '6':
+            playercount = '6v6'
+        case '7':
+            playercount = '7v7'
+        case '8':
+            playercount = '8v8'
+        case '9':
+            playercount = '9v9'
+        case '10':
+            playercount = '10v10'
+        case _:
+            playercount = '2v2'
+
+    #pull the maps
+    try:
+        service = build('sheets', 'v4', credentials=creds)
+
+        # Call the Sheets API
+        sheet = service.spreadsheets()
+        result = sheet.values().get(spreadsheetId=SAMPLE_SPREADSHEET_ID,
+                                    range=SAMPLE_RANGE_NAME).execute()
+        values = result.get('values', [])
+
+        #if it's empty say so
+        if not values:
+            print('No data found.')
+            return
+
+        #pull map names from column A (0) based on team sizes in column C (2)
+        for row in values:
+            if playercount in row[2]:
+                pulledmaps.append(row[0])
+
+    except HttpError as err:
+        print(err)
+
+    #print(pulledmaps)
+    return pulledmaps
+
+#if __name__ == '__main__':
+    #main()
+
+load_dotenv()
+TOKEN = os.getenv('DISCORD_TOKEN')
+
+bot = commands.Bot(command_prefix='!')
+
+    #TODO: track how many players have done !add and give hints about next steps (!maps2 or add more or !maps3 etc)
+    #TODO: automatically newmaps
+    #TODO: some kind of vote tracking/closure system?
+
+@bot.command(name='maps', help='Returns 3 random maps from neon\'s spreadsheet, use !maps 3 for 3v3, !maps 4 for 4v4 etc')
+async def choose_maps(ctx, players):
+    #populate the maplist from google sheets
+    maplistsheets = pullmaps(players)
+
+    #get 3 at random
+    #TODO: make this less than fully random? remember previous pick(s)?
+    mapstochoose = [random.sample(maplistsheets,3)]
+    print(mapstochoose)
+
+    #format response
+    #TODO: update results with names of voters?
+    #TODO: automatically newmaps at (playercount) total votes?
+    response = ( '```'
+        '1    ' + mapstochoose[0][0] + '\n'
+        '2    ' + mapstochoose[0][1] + '\n'
+        '3    ' + mapstochoose[0][2] + '\n'
+        '4    newmaps```'
+    )
+    message = await ctx.send(response)
+
+    #add reactions for vote tracking
+    emojis = ['1️⃣', '2️⃣', '3️⃣', '4️⃣']
+    for emoji in emojis:
+        await message.add_reaction(emoji)
+
+
+#if something bad happens write it down
+@bot.event
+async def on_error(event, *args, **kwargs):
+    with open('err.log', 'a') as f:
+        if event == 'on_message':
+            f.write(f'Unhandled message: {args[0]}\n')
+        else:
+            raise
+
+#lemme know when it's ready
+@bot.event
+async def on_ready():
+    print(f'{bot.user.name} has connected to Discord!')
+
+bot.run(TOKEN)