brainsqueeze commited on
Commit
3336f1b
·
verified ·
1 Parent(s): e1fc32c

allow bypass of additional org attributes for search

Browse files
Files changed (1) hide show
  1. common/org_search_component.py +21 -18
common/org_search_component.py CHANGED
@@ -17,7 +17,7 @@ class OrgSearch(BaseAPI):
17
  headers={"x-api-key": os.getenv('CDS_API_KEY')}
18
  )
19
 
20
- def __call__(self, name: str, **kwargs):
21
  is_valid = False
22
 
23
  payload = {
@@ -28,24 +28,27 @@ class OrgSearch(BaseAPI):
28
  "status": "authorized"
29
  }
30
 
31
- if kwargs.get("ein"):
32
- ein = kwargs.get("ein")
33
- if "-" not in ein:
34
- ein = f"{ein[:2]}-{ein[2:]}"
35
- payload["ids"] = [{
36
- "value": ein,
37
- "type": "ein"
38
- }]
39
- is_valid = True
40
-
41
- if kwargs.get("street") or kwargs.get("city") or kwargs.get("state") or kwargs.get("postal_code"):
42
- payload["addresses"] = [{
43
- "street1": kwargs.get("street") or "",
44
- "city": kwargs.get("city") or "",
45
- "state": kwargs.get("state") or "",
46
- "postal_code": kwargs.get("postal_code") or ""
47
- }]
48
  is_valid = True
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
49
 
50
  if not is_valid:
51
  return None
 
17
  headers={"x-api-key": os.getenv('CDS_API_KEY')}
18
  )
19
 
20
+ def __call__(self, name: str, name_only: bool = False, **kwargs):
21
  is_valid = False
22
 
23
  payload = {
 
28
  "status": "authorized"
29
  }
30
 
31
+ if name_only:
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
32
  is_valid = True
33
+ else:
34
+ if kwargs.get("ein"):
35
+ ein = kwargs.get("ein")
36
+ if "-" not in ein:
37
+ ein = f"{ein[:2]}-{ein[2:]}"
38
+ payload["ids"] = [{
39
+ "value": ein,
40
+ "type": "ein"
41
+ }]
42
+ is_valid = True
43
+
44
+ if kwargs.get("street") or kwargs.get("city") or kwargs.get("state") or kwargs.get("postal_code"):
45
+ payload["addresses"] = [{
46
+ "street1": kwargs.get("street") or "",
47
+ "city": kwargs.get("city") or "",
48
+ "state": kwargs.get("state") or "",
49
+ "postal_code": kwargs.get("postal_code") or ""
50
+ }]
51
+ is_valid = True
52
 
53
  if not is_valid:
54
  return None