Create sheets.py

sheets.py

@@ -0,0 +1,54 @@
+import gspread
+from oauth2client.service_account import ServiceAccountCredentials
+
+def get_sheet():
+    scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
+    creds = ServiceAccountCredentials.from_json_keyfile_name("credentials.json", scope)
+    client = gspread.authorize(creds)
+    sheet = client.open("Leads").sheet1
+    return sheet
+
+import requests
+from bs4 import BeautifulSoup
+
+def fetch_google_search_results(query, num_results):
+    headers = {
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"}
+    search_url = f"https://www.google.com/search?q={query}&num={num_results}"
+    response = requests.get(search_url, headers=headers)
+    response.raise_for_status()
+    return response.text
+
+def extract_company_info(html_content):
+    soup = BeautifulSoup(html_content, 'html.parser')
+    results = soup.find_all('div', class_='tF2Cxc')
+    data = []
+    for result in results:
+        company_info = {}
+        company_info['Name'] = result.find('h3').text if result.find('h3') else ''
+        company_info['Website'] = result.find('a')['href'] if result.find('a') else ''
+        company_info['Additional Info'] = result.find('span', class_='aCOpRe').text if result.find('span', class_='aCOpRe') else ''
+        data.append(company_info)
+    return data
+
+def insert_data(sheet, data):
+    existing_records = sheet.get_all_records()
+    existing_emails = [record['Email'] for record in existing_records]
+
+    for entry in data:
+        if entry['Email'] not in existing_emails:
+            row = [entry['Name'], entry.get('Email', ''), entry.get('Mobile Number', ''), entry.get('POC', ''), entry['Website'], entry.get('LinkedIn Profile', ''), 'Not Contacted']
+            sheet.append_row(row)
+
+def remove_duplicates(sheet):
+    records = sheet.get_all_records()
+    unique_records = []
+    emails = set()
+    for record in records:
+        if record['Email'] not in emails:
+            unique_records.append(record)
+            emails.add(record['Email'])
+    sheet.clear()
+    sheet.append_row(["Name", "Email", "Mobile Number", "POC", "Website", "LinkedIn Profile", "Status"])
+    for record in unique_records:
+        sheet.append_row([record['Name'], record['Email'], record['Mobile Number'], record['POC'], record['Website'], record['LinkedIn Profile'], record['Status']])