zliang commited on
Commit
695cdfe
1 Parent(s): 61a3301

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +7 -3
main.py CHANGED
@@ -9,11 +9,15 @@ app = FastAPI()
9
  BASE_URL = 'https://doi.org/'
10
 
11
  @app.get("/title2ref/")
12
- async def ttile2doi(title:str):
13
  cr = Crossref()
14
 
15
- result = cr.works(query = "title")
16
- doi = result['message']['items'][0]['DOI']
 
 
 
 
17
  url = BASE_URL + doi
18
  req = urllib.request.Request(url)
19
  req.add_header('Accept', 'application/x-bibtex')
 
9
  BASE_URL = 'https://doi.org/'
10
 
11
  @app.get("/title2ref/")
12
+ async def title2doi(title:str):
13
  cr = Crossref()
14
 
15
+ result = cr.works(query = title)
16
+ for i, item in enumerate(result['message']['items']):
17
+ if 'group-title' in item and item['group-title'] != 'Preprints':
18
+ doi = result['message']['items'][i-1]['DOI']
19
+ break
20
+
21
  url = BASE_URL + doi
22
  req = urllib.request.Request(url)
23
  req.add_header('Accept', 'application/x-bibtex')