Skip to content

Commit aa82ab6

Browse files
#REQ_DEV
Prompt Category: Requirement Development Prompt: Suggest Python code to connect to the Source google sheet and extract the required columns -Email address - Tool used - Feature Used - Context Awareness - Autonomy - Experience - Output Quality - Overall Rating - Unique ID - Pod User Observation: A python file extract_data.py was created Response ID: 01779e1d-1d40-4202-9851-011953b19ff0
1 parent a2b408b commit aa82ab6

File tree

1 file changed

+61
-0
lines changed

1 file changed

+61
-0
lines changed

extract_data.py

+61
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,61 @@
1+
from google.oauth2.service_account import Credentials
2+
from googleapiclient.discovery import build
3+
import pandas as pd
4+
5+
def get_google_sheet_data():
6+
# Define the scope and credentials
7+
SCOPES = ['https://www.googleapis.com/auth/spreadsheets.readonly']
8+
creds = Credentials.from_service_account_file('credentials.json', scopes=SCOPES)
9+
10+
# Create the service
11+
service = build('sheets', 'v4', credentials=creds)
12+
13+
# Spreadsheet ID from the URL
14+
SPREADSHEET_ID = '15FMeidgU2Dg7Q4JKPkLAdJmQ3IxWCWJXjhCo9UterCE'
15+
RANGE_NAME = 'POD 5!A1:CE1000' # We'll fetch all columns and filter later
16+
17+
try:
18+
# Call the Sheets API
19+
sheet = service.spreadsheets()
20+
result = sheet.values().get(
21+
spreadsheetId=SPREADSHEET_ID,
22+
range=RANGE_NAME
23+
).execute()
24+
25+
values = result.get('values', [])
26+
27+
if not values:
28+
print('No data found.')
29+
return None
30+
31+
# Convert to DataFrame
32+
df = pd.DataFrame(values[1:], columns=values[0]) # First row as headers
33+
34+
# Select required columns
35+
required_columns = [
36+
'Email address',
37+
'Tool used',
38+
'Feature Used',
39+
'Context Awareness',
40+
'Autonomy',
41+
'Experience',
42+
'Output Quality',
43+
'Overall Rating',
44+
'Unique ID',
45+
'Pod'
46+
]
47+
48+
# Filter only required columns
49+
filtered_df = df[required_columns]
50+
51+
return filtered_df
52+
53+
except Exception as e:
54+
print(f"An error occurred: {e}")
55+
return None
56+
57+
if __name__ == "__main__":
58+
data = get_google_sheet_data()
59+
if data is not None:
60+
print("Successfully retrieved data:")
61+
print(data.head()) # Display first 5 rows

0 commit comments

Comments
 (0)