From 78a6599d822b019c72341b616ea39bf7219b87dd Mon Sep 17 00:00:00 2001
From: Griffiths Lott <glott@leafnow.com>
Date: Wed, 21 Dec 2022 15:53:10 -0500
Subject: [PATCH] Added conversion methods to EINData. Added option to return
 search_eins as a dataframe

---
 EINService.py | 24 ++++++++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/EINService.py b/EINService.py
index 25f2a4b..2bd88ea 100644
--- a/EINService.py
+++ b/EINService.py
@@ -4,7 +4,8 @@ import re
 from abc import ABC, abstractmethod
 from dataclasses import dataclass
 from logging import warning, debug, error
-from pandas import DataFrame
+from pandas import DataFrame, Series
+import pandas as pd
 
 
 SAMPLE_EIN = "59-1571026"
@@ -75,6 +76,18 @@ class EINData:
         compareDict["score"] = score
         return compareDict
 
+    def as_dict(self) -> dict:
+        return {"BusinessName": self.buinessName, "Address1": self.address1, 
+        "City":self.city, "State": self.state, "Zip": self.zip, "Phone": self.phone}
+
+
+    def as_series(self) -> Series:
+        """
+        Converts object into a pandas Series
+        """
+        return Series(self.to_dict())
+    
+    
 
 def dataframe_to_eins(df: DataFrame, 
     einLabel: str = "Lessee Tax-ID",
@@ -199,9 +212,12 @@ class EINTaxIDService(__EINService):
             data["phone"]
         )
 
-    def search_eins(self, eins: str) -> list[EINData]:
-        return [self.search_ein(ein) for ein in eins]
-
+    def search_eins(self, eins: str, asDataFrame: bool = False) -> list[EINData]:
+        if asDataFrame:
+            dataList =[self.search_ein(ein).as_series() for ein in eins]
+            return pd.concat(dataList, ignore_index=True)
+        else:
+            return [self.search_ein(ein) for ein in eins]          
 
     def _parse_return(self, content: str) -> dict:
         """