Coverage for sample_project/querysets.py: 0%

1import os 

2 

3from apis_core.utils.autocomplete import ( 

4 ExternalAutocomplete, 

5 LobidAutocompleteAdapter, 

6 TypeSenseAutocompleteAdapter, 

7) 

8 

9 

10class GroupExternalAutocomplete(ExternalAutocomplete): 

11 adapters = [ 

12 LobidAutocompleteAdapter( 

13 params={ 

14 "filter": "type:CorporateBody", 

15 "format": "json:preferredName,geographicAreaCode,dateOfEstablishment,broaderTermInstantial", 

16 } 

17 ), 

18 ] 

19 

20 

21class PlaceExternalAutocomplete(ExternalAutocomplete): 

22 adapters = [ 

23 TypeSenseAutocompleteAdapter( 

24 collections=[ 

25 "prosnet-wikidata-place-index", 

26 "prosnet-geonames-place-index", 

27 ], 

28 token=os.getenv("TYPESENSE_TOKEN", None), 

29 server=os.getenv("TYPESENSE_SERVER", None), 

30 ), 

31 LobidAutocompleteAdapter( 

32 params={ 

33 "filter": "type:PlaceOrGeographicName", 

34 "format": "json:preferredName", 

35 } 

36 ), 

37 ] 

38 

39 

40class PersonExternalAutocomplete(ExternalAutocomplete): 

41 adapters = [ 

42 TypeSenseAutocompleteAdapter( 

43 collections="prosnet-wikidata-person-index", 

44 token=os.getenv("TYPESENSE_TOKEN", None), 

45 server=os.getenv("TYPESENSE_SERVER", None), 

46 ), 

47 LobidAutocompleteAdapter( 

48 params={ 

49 "filter": "type:Person", 

50 "format": "json:preferredName,professionOrOccupation", 

51 } 

52 ), 

53 ]