From d3dd81ab000b9bcb7cb91ba8586dadd35c63dc70 Mon Sep 17 00:00:00 2001 From: tsb1995 <47466105+tsb1995@users.noreply.github.com> Date: Sun, 7 Jun 2020 10:40:41 -0700 Subject: [PATCH] Cleaned up ZillowGrabber code, removed Test and Credentials --- ZillowGrabber.py | 100 +++++++++++++++++++++++------------------------ credentials.json | 1 - 2 files changed, 49 insertions(+), 52 deletions(-) delete mode 100644 credentials.json diff --git a/ZillowGrabber.py b/ZillowGrabber.py index 371d140..8b5843e 100644 --- a/ZillowGrabber.py +++ b/ZillowGrabber.py @@ -76,64 +76,62 @@ range_ = "Sheet1!A:A" d = datetime.datetime.today() sheet_name = 'Housing ' + d.strftime('%d-%m-%Y') -def main(): - creds = None - # The file token.pickle 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.pickle'): - with open('token.pickle', 'rb') as token: - creds = pickle.load(token) - # 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.pickle', 'wb') as token: - pickle.dump(creds, token) +creds = None +# The file token.pickle 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.pickle'): + with open('token.pickle', 'rb') as token: + creds = pickle.load(token) - service = build('sheets', 'v4', credentials=creds) +# 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.pickle', 'wb') as token: + pickle.dump(creds, token) - # Call the Sheets API - sheet = service.spreadsheets() +service = build('sheets', 'v4', credentials=creds) - # Create new Sheet - spreadsheet = { - 'properties': { - 'title': sheet_name - } +# Call the Sheets API +sheet = service.spreadsheets() + +# Create new Sheet +spreadsheet = { + 'properties': { + 'title': sheet_name } - spreadsheet = service.spreadsheets().create(body=spreadsheet, - fields='spreadsheetId').execute() - spreadsheet_id = spreadsheet.get('spreadsheetId') - print('Spreadsheet ID: {0}'.format(spreadsheet.get('spreadsheetId'))) +} +spreadsheet = service.spreadsheets().create(body=spreadsheet, + fields='spreadsheetId').execute() +spreadsheet_id = spreadsheet.get('spreadsheetId') +print('Spreadsheet ID: {0}'.format(spreadsheet.get('spreadsheetId'))) - # Fill in header - list = [df.columns.tolist()] +# Fill in header +list = [df.columns.tolist()] - resource = { - "majorDimension": "ROWS", - "values": list - } +resource = { + "majorDimension": "ROWS", + "values": list +} - request = service.spreadsheets().values().append(spreadsheetId=spreadsheet_id, range=range_, body=resource, - valueInputOption="USER_ENTERED") - response = request.execute() +request = service.spreadsheets().values().append(spreadsheetId=spreadsheet_id, range=range_, body=resource, + valueInputOption="USER_ENTERED") +response = request.execute() - # Fill in body - list = df.values.tolist() +# Fill in body +list = df.values.tolist() - resource = { - "majorDimension": "ROWS", - "values": list - } +resource = { + "majorDimension": "ROWS", + "values": list +} - request = service.spreadsheets().values().append(spreadsheetId=spreadsheet_id, range=range_, body=resource, - valueInputOption="USER_ENTERED") - response = request.execute() - -main() +request = service.spreadsheets().values().append(spreadsheetId=spreadsheet_id, range=range_, body=resource, + valueInputOption="USER_ENTERED") +response = request.execute() diff --git a/credentials.json b/credentials.json deleted file mode 100644 index a0d6bc7..0000000 --- a/credentials.json +++ /dev/null @@ -1 +0,0 @@ -{"installed":{"client_id":"495919805073-o2o41tu9c9pv9mpo4ugjdcca6h5jg5u7.apps.googleusercontent.com","project_id":"zillowfun-1581659411719","auth_uri":"https://accounts.google.com/o/oauth2/auth","token_uri":"https://oauth2.googleapis.com/token","auth_provider_x509_cert_url":"https://www.googleapis.com/oauth2/v1/certs","client_secret":"HL2xQvsWITylqar1GsC8noW1","redirect_uris":["urn:ietf:wg:oauth:2.0:oob","http://localhost"]}} \ No newline at end of file