-
Notifications
You must be signed in to change notification settings - Fork 0
/
sheets.py
45 lines (37 loc) · 1.37 KB
/
sheets.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
import os
from googleapiclient.discovery import build
from httplib2 import Http
from oauth2client import file, client, tools
SCOPES = 'https://www.googleapis.com/auth/spreadsheets'
def authenticate():
store = file.Storage('token.json')
creds = store.get()
if not creds or creds.invalid:
flow = client.flow_from_clientsecrets('credentials.json', SCOPES)
creds = tools.run_flow(flow, store)
return build('sheets', 'v4', http=creds.authorize(Http()))
service = authenticate()
def get_item_codes(spreadsheet_id):
range_name = 'Alliance project!A2:A'
result = service.spreadsheets().values().get(spreadsheetId=spreadsheet_id,
range=range_name).execute()
values = result.get('values', [])
item_codes = []
if not values:
print('No data found.')
else:
for row in values:
if row:
code = row[0]
item_codes.append(code)
print('Reading {} item codes from spreadsheet\n'.format(len(item_codes)))
return item_codes
def write_to_sheet(values, spreadsheet_id):
range_name = 'Alliance project!C2:G'
body = {
'values': values,
'majorDimension': 'ROWS',
}
result = service.spreadsheets().values().update(
spreadsheetId=spreadsheet_id, range=range_name,
valueInputOption='RAW', body=body).execute()