Files
RushStatistics/loader/sheetloader.py
2024-06-08 21:33:53 -07:00

32 lines
943 B
Python

import os
import requests
import sys
DOCUMENT_ID = '1VKujaowUSxB9SuBdMt81aFlUGWo2fUCAOzX2PjYQVxs'
GAME_SHEET_ID = '160661246'
CHARACTTER_SHEET_ID = '1445780435'
def getGoogleSheet(spreadsheetId, sheetId, outFile):
url = f'https://docs.google.com/spreadsheets/d/{spreadsheetId}/export?gid={sheetId}&format=csv'
response = requests.get(url)
if response.status_code == 200:
filepath = os.path.join('./', outFile)
with open(filepath, 'wb') as f:
f.write(response.content)
print('CSV file saved to: {}'.format(filepath))
else:
print(f'Error downloading sheet: {response.status_code}')
sys.exit(1)
def downloadGamesCSV(outFile):
getGoogleSheet(DOCUMENT_ID, GAME_SHEET_ID, outFile)
def downloadCharactersCSV(outFile):
getGoogleSheet(DOCUMENT_ID, CHARACTTER_SHEET_ID, outFile)
if __name__ == "__main__":
getGoogleSheet('1VKujaowUSxB9SuBdMt81aFlUGWo2fUCAOzX2PjYQVxs', '1445780435', './here', 'potato')