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

1import os 6 ctx1abcdef

2from typing import Optional 6 ctx1abcdef

3 

4from mongodb_odm import ( 6 ctx1abcdef

5 ASCENDING, 

6 Document, 

7 Field, 

8 IndexModel, 

9 ODMObjectId, 

10 Relationship, 

11 apply_indexes, 

12 connect, 

13) 

14 

15 

16class Country(Document): 6 ctx1abcdef

17 name: str 6 ctx1abcdef

18 

19 

20class Skill(Document): 6 ctx1abcdef

21 name: str 6 ctx1abcdef

22 

23 

24class Player(Document): 6 ctx1abcdef

25 name: str 6 ctx1abcdef

26 country_id: ODMObjectId 6 ctx1abcdef

27 

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

29 

30 class ODMConfig(Document.ODMConfig): 6 ctx1abcdef

31 indexes = [ 6 ctx1abcdef

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

33 ] 

34 

35 

36class PlayerSkill(Document): 6 ctx1abcdef

37 player_id: ODMObjectId 6 ctx1abcdef

38 skill_id: ODMObjectId 6 ctx1abcdef

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

40 

41 player: Optional[Player] = Relationship(local_field="player_id") 6 ctx1abcdef

42 skill: Optional[Skill] = Relationship(local_field="skill_id") 6 ctx1abcdef

43 

44 class ODMConfig(Document.ODMConfig): 6 ctx1abcdef

45 indexes = [ 6 ctx1abcdef

46 IndexModel( 

47 [("player_id", ASCENDING), ("skill_id", ASCENDING)], 

48 unique=True, 

49 ), 

50 ] 

51 

52 

53def insert_data(): 6 ctx1abcdef

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

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

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

57 

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

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

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

61 

62 pele = Player(name="Pelé", country_id=brazil.id).create() 6 ctx1abcdef

63 PlayerSkill(player_id=pele.id, skill_id=catching.id, rating=49).create() 6 ctx1abcdef

64 PlayerSkill(player_id=pele.id, skill_id=kicking.id, rating=49).create() 6 ctx1abcdef

65 

66 maradona = Player(name="Diego Maradona", country_id=argentina.id).create() 6 ctx1abcdef

67 PlayerSkill(player_id=maradona.id, skill_id=catching.id, rating=48).create() 6 ctx1abcdef

68 PlayerSkill(player_id=maradona.id, skill_id=kicking.id, rating=49).create() 6 ctx1abcdef

69 

70 zidane = Player(name="Zinedine Zidane", country_id=france.id).create() 6 ctx1abcdef

71 PlayerSkill(player_id=zidane.id, skill_id=running.id, rating=42).create() 6 ctx1abcdef

72 PlayerSkill(player_id=zidane.id, skill_id=kicking.id, rating=42).create() 6 ctx1abcdef

73 

74 

75def read_data(): 6 ctx1abcdef

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

77 skills = list(PlayerSkill.find({"player_id": pele.id})) 6 ctx1abcdef

78 

79 print(pele) 6 ctx1abcdef

80 print(skills) 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()