Coverage for docs_src / tutorial / relation / many_to_many_relationship / tutorial001.py: 100%

1import os 6 ctx1abcdef

2from typing import Optional 6 ctx1abcdef

3 

4from mongodb_odm import ( 6 ctx1abcdef

5 ASCENDING, 

6 BaseModel, 

7 Document, 

8 Field, 

9 IndexModel, 

10 ODMObjectId, 

11 Relationship, 

12 apply_indexes, 

13 connect, 

14) 

15 

16 

17class Country(Document): 6 ctx1abcdef

18 name: str 6 ctx1abcdef

19 

20 

21class Skill(Document): 6 ctx1abcdef

22 name: str 6 ctx1abcdef

23 

24 

25class EmbeddedSkill(BaseModel): 6 ctx1abcdef

26 skill_id: ODMObjectId 6 ctx1abcdef

27 rating: int = Field(default=0, ge=1, le=100) 6 ctx1abcdef

28 

29 

30class Player(Document): 6 ctx1abcdef

31 name: str 6 ctx1abcdef

32 country_id: ODMObjectId 6 ctx1abcdef

33 skills: list[EmbeddedSkill] = [] 6 ctx1abcdef

34 

35 country: Optional[Country] = Relationship(local_field="country_id") 6 ctx1abcdef

36 

37 class ODMConfig(Document.ODMConfig): 6 ctx1abcdef

38 indexes = [ 6 ctx1abcdef

39 IndexModel([("country_id", ASCENDING)]), 

40 ] 

41 

42 

43def insert_data(): 6 ctx1abcdef

44 brazil = Country(name="Brazil").create() 6 ctx1abcdef

45 argentina = Country(name="Argentina").create() 6 ctx1abcdef

46 france = Country(name="France").create() 6 ctx1abcdef

47 

48 catching = Skill(name="Catching") 6 ctx1abcdef

49 running = Skill(name="Running") 6 ctx1abcdef

50 kicking = Skill(name="Kicking") 6 ctx1abcdef

51 

52 Player( 6 ctx1abcdef

53 name="Pelé", 

54 country_id=brazil.id, 

55 skills=[ 

56 EmbeddedSkill(skill_id=catching.id, rating=49), 

57 EmbeddedSkill(skill_id=kicking.id, rating=49), 

58 ], 

59 ).create() 

60 Player( 6 ctx1abcdef

61 name="Diego Maradona", 

62 country_id=argentina.id, 

63 skills=[ 

64 EmbeddedSkill(skill_id=catching.id, rating=48), 

65 EmbeddedSkill(skill_id=kicking.id, rating=49), 

66 ], 

67 ).create() 

68 Player( 6 ctx1abcdef

69 name="Zinedine Zidane", 

70 country_id=france.id, 

71 skills=[ 

72 EmbeddedSkill(skill_id=running.id, rating=42), 

73 EmbeddedSkill(skill_id=kicking.id, rating=42), 

74 ], 

75 ).create() 

76 

77 

78def read_data(): 6 ctx1abcdef

79 pele = Player.get({Player.name: "Pelé"}) 6 ctx1abcdef

80 print(pele) 6 ctx1abcdef

81 

82 

83def configuration(): 6 ctx1abcdef

84 connect(os.environ.get("MONGO_URL", "mongodb://localhost:27017/testdb")) 6 ctx1abcdef

85 apply_indexes() 6 ctx1abcdef

86 

87 

88def main(): 6 ctx1abcdef

89 configuration() 6 ctx1abcdef

90 insert_data() 6 ctx1abcdef

91 read_data() 6 ctx1abcdef

92 

93 

94if __name__ == "__main__": 6 ctx1abcdef

95 main()