Coverage for faststream / specification / asyncapi / v2_6_0 / schema / contact.py: 89%

1from typing import Any, cast, overload (empty)

2 

3from pydantic import AnyHttpUrl, BaseModel (empty)

4from typing_extensions import Self (empty)

5 

6from faststream._internal._compat import PYDANTIC_V2, EmailStr (empty)

7from faststream._internal.utils.data import filter_by_dict (empty)

8from faststream.specification.schema.extra import ( (empty)

9 Contact as SpecContact, 

10 ContactDict, 

11) 

12 

13 

14class Contact(BaseModel): (empty)

15 """A class to represent a contact. 

16 

17 Attributes: 

18 name : name of the contact (str) 

19 url : URL of the contact (Optional[AnyHttpUrl]) 

20 email : email of the contact (Optional[EmailStr]) 

21 """ 

22 

23 name: str (empty)

24 # Use default values to be able build from dict 

25 url: AnyHttpUrl | None = None (empty)

26 email: EmailStr | None = None (empty)

27 

28 if PYDANTIC_V2: 28 ↛ 33line 28 didn't jump to line 33 because the condition on line 28 was always true(empty)

29 model_config = {"extra": "allow"} (empty)

30 

31 else: 

32 

33 class Config: 

34 extra = "allow" 

35 

36 @overload (empty)

37 @classmethod (empty)

38 def from_spec(cls, contact: None) -> None: ... (empty)

39 

40 @overload (empty)

41 @classmethod (empty)

42 def from_spec(cls, contact: SpecContact) -> Self: ... (empty)

43 

44 @overload (empty)

45 @classmethod (empty)

46 def from_spec(cls, contact: ContactDict) -> Self: ... (empty)

47 

48 @overload (empty)

49 @classmethod (empty)

50 def from_spec(cls, contact: dict[str, Any]) -> dict[str, Any]: ... (empty)

51 

52 @classmethod (empty)

53 def from_spec( (empty)

54 cls, 

55 contact: SpecContact | ContactDict | dict[str, Any] | None, 

56 ) -> Self | dict[str, Any] | None: 

57 if contact is None: (empty)

58 return None (empty)

59 

60 if isinstance(contact, SpecContact): (empty)

61 return cls( (empty)

62 name=contact.name, 

63 url=contact.url, 

64 email=contact.email, 

65 ) 

66 

67 contact = cast("dict[str, Any]", contact) (empty)

68 contact_data, custom_data = filter_by_dict(ContactDict, contact) (empty)

69 

70 if custom_data: (empty)

71 return contact (empty)

72 

73 return cls(**contact_data) (empty)