feat: add chrome headers for avoiding blocking

scraper.py

@@ -1,3 +1,5 @@
+from unicodedata import name
+import pandas as pd
 import requests
 import re
 import csv
@@ -7,7 +9,12 @@ from bs4 import BeautifulSoup
 sec_url = 'https://www.sec.gov'
 
 def get_request(url):
-    return requests.get(url)
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/97.0.4692.99 Safari/537.36',
+        'Accept-Encoding': 'gzip, deflate, br',
+        'HOST': 'www.sec.gov',
+    }
+    return requests.get(url, headers=headers)
 
 def create_url(cik):
     return 'https://www.sec.gov/cgi-bin/browse-edgar?CIK={}&owner=exclude&action=getcompany&type=13F-HR'.format(cik)
@@ -16,29 +23,66 @@ def get_user_input():
     cik = input("Enter 10-digit CIK number: ")
     return cik
 
+
+def scrap_company_report(requested_cik):
+    # Find mutual fund by CIK number on EDGAR
+    response = get_request(create_url(requested_cik))
+    soup = BeautifulSoup(response.text, "html.parser")
+    tags = soup.findAll('a', id="documentsbutton")
+
+    last_report = (sec_url + tags[0]['href'])
+    previous_report = (sec_url + tags[1]['href'])
+    scrap_report_by_url(last_report, "last_report")
+    scrap_report_by_url(previous_report, "previous_report")
+
+
+def scrap_report_by_url(url, filename):
+    response_two = get_request(url)
+    soup_two = BeautifulSoup(response_two.text, "html.parser")
+    tags_two = soup_two.findAll('a', attrs={'href': re.compile('xml')})
+    xml_url = tags_two[3].get('href')
+
+    response_xml = get_request(sec_url + xml_url)
+    soup_xml = BeautifulSoup(response_xml.content, "lxml")
+    xml_to_csv(soup_xml, filename)
+
+
+def xml_to_csv(soup_xml, name):
+
+    columns = [
+        "Name of Issuer",
+        "CUSIP",
+        "Value (x$1000)",
+        "Shares",
+        "Investment Discretion",
+        "Voting Sole / Shared / None"
+    ]
+    issuers = soup_xml.body.findAll(re.compile('nameofissuer'))
+    cusips = soup_xml.body.findAll(re.compile('cusip'))
+    values = soup_xml.body.findAll(re.compile('value'))
+    sshprnamts = soup_xml.body.findAll('sshprnamt')
+    sshprnamttypes = soup_xml.body.findAll(re.compile('sshprnamttype'))
+    investmentdiscretions = soup_xml.body.findAll(re.compile('investmentdiscretion'))
+    soles = soup_xml.body.findAll(re.compile('sole'))
+    shareds = soup_xml.body.findAll(re.compile('shared'))
+    nones = soup_xml.body.findAll(re.compile('none'))
+
+    df = pd.DataFrame(columns= columns)
+
+    for issuer, cusip, value, sshprnamt, sshprnamttype, investmentdiscretion, sole, shared, none in zip(issuers, cusips, values, sshprnamts, sshprnamttypes, investmentdiscretions, soles, shareds, nones):
+        row = {
+            "Name of Issuer": issuer.text,
+            "CUSIP": cusip.text,
+            "Value (x$1000)": value.text,
+            "Shares": f"{sshprnamt.text} {sshprnamttype.text}",
+            "Investment Discretion": investmentdiscretion.text,
+            "Voting Sole / Shared / None": f"{sole.text} / {shared.text} / {none.text}"
+        }
+        df = df.append(row, ignore_index=True)
+
+
+    df.to_csv(f"{name}.csv")
+
+
 requested_cik = get_user_input()
-
-# Find mutual fund by CIK number on EDGAR
-response = get_request(create_url(requested_cik))
-soup = BeautifulSoup(response.text, "html.parser")
-tags = soup.findAll('a', id="documentsbutton")
-
-# Find latest 13F report for mutual fund
-response_two = get_request(sec_url + tags[0]['href'])
-soup_two = BeautifulSoup(response_two.text, "html.parser")
-tags_two = soup_two.findAll('a', attrs={'href': re.compile('xml')})
-xml_url = tags_two[3].get('href')
-
-response_xml = get_request(sec_url + xml_url)
-soup_xml = BeautifulSoup(response_xml.content, "lxml")
-
-# Find all issuers
-issuers = soup_xml.body.findAll(re.compile('nameofissuer'))
-for issuer in issuers:
-    print(issuer.text)
-
-# Write issuer names to TSV file
-with open('{}.tsv'.format(requested_cik), 'wt') as out_file:
-    tsv_writer = csv.writer(out_file, delimiter='\t')
-    for issuer in issuers:
-        tsv_writer.writerow([issuer.text])
+scrap_company_report(requested_cik)