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 data_mapping={"label": "label"}, 

18 ), 

19 ] 

20 

21 

22class PlaceExternalAutocomplete(ExternalAutocomplete): 

23 adapters = [ 

24 TypeSenseAutocompleteAdapter( 

25 collections=[ 

26 "prosnet-wikidata-place-index", 

27 "prosnet-geonames-place-index", 

28 ], 

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

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

31 data_mapping={"label": ["document", "label"]}, 

32 ), 

33 LobidAutocompleteAdapter( 

34 params={ 

35 "filter": "type:PlaceOrGeographicName", 

36 "format": "json:preferredName", 

37 }, 

38 data_mapping={"label": "label"}, 

39 ), 

40 ] 

41 

42 

43class PersonExternalAutocomplete(ExternalAutocomplete): 

44 adapters = [ 

45 TypeSenseAutocompleteAdapter( 

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

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

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

49 ), 

50 LobidAutocompleteAdapter( 

51 params={ 

52 "filter": "type:Person", 

53 "format": "json:preferredName,professionOrOccupation", 

54 }, 

55 data_mapping={"surname": "label"}, 

56 ), 

57 ]