Skip to content

Commit 987a91a

Browse files
author
Roman Inflianskas
committed
Add servers
Test results were updated. Return value typing was fixed along the way, as it wasn't correct.
1 parent 6a0c79f commit 987a91a

6 files changed

Lines changed: 11 additions & 2 deletions

File tree

fastapi_code_generator/parser.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -236,8 +236,12 @@ def __init__(
236236
self.imports_for_fastapi: Imports = Imports()
237237
self.data_types: List[DataType] = []
238238

239-
def parse_info(self) -> Optional[List[Dict[str, List[str]]]]:
240-
return self.raw_obj.get('info')
239+
def parse_info(self) -> Optional[Dict[str, Any]]:
240+
result = self.raw_obj.get('info', {}).copy()
241+
servers = self.raw_obj.get('servers')
242+
if servers:
243+
result['servers'] = servers
244+
return result or None
241245

242246
def parse_parameters(self, parameters: ParameterObject, path: List[str]) -> None:
243247
super().parse_parameters(parameters, path)

tests/data/expected/openapi/default_template/body_and_parameters/main.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
title='Swagger Petstore',
2424
license={'name': 'MIT'},
2525
description='This description is for testing\nmulti-line\ndescription\n',
26+
servers=[{'url': 'http://petstore.swagger.io/v1'}],
2627
)
2728

2829

tests/data/expected/openapi/default_template/content_in_parameters/main.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
title='Swagger Petstore',
1616
license={'name': 'MIT'},
1717
description='This description is for testing\nmulti-line\ndescription\n',
18+
servers=[{'url': 'http://petstore.swagger.io/v1'}],
1819
)
1920

2021

tests/data/expected/openapi/default_template/oas_response_reference/main.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
title='Swagger Petstore',
1616
license={'name': 'MIT'},
1717
description='This description is for testing\nmulti-line\ndescription\n',
18+
servers=[{'url': 'http://petstore.swagger.io/v1'}],
1819
)
1920

2021

tests/data/expected/openapi/default_template/simple/main.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
title='Swagger Petstore',
1616
license={'name': 'MIT'},
1717
description='This description is for testing\nmulti-line\ndescription\n',
18+
servers=[{'url': 'http://petstore.swagger.io/v1'}],
1819
)
1920

2021

tests/data/expected/openapi/remote_ref/body_and_parameters/main.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
title='Swagger Petstore',
1616
license={'name': 'MIT'},
1717
description='This description is for testing\nmulti-line\ndescription\n',
18+
servers=[{'url': 'http://petstore.swagger.io/v1'}],
1819
)
1920

2021

0 commit comments

Comments
 (0)