-
Notifications
You must be signed in to change notification settings - Fork 0
/
Journal.py
82 lines (70 loc) · 3.01 KB
/
Journal.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
import sys
import json
from Training import *
class Journal( object ):
"""Journal is a collection of trainings"""
def __init__( self, trainings_list, journal_description_dict ):
self.trainings = trainings_list
self.description = journal_description_dict
def add_training( self, training ):
"""Add single training. Only one is expected."""
self.trainings.append( training )
def add_description( self, description_dict ):
self.description.update( description_dict )
def __repr__( self ):
return "Journal: \n" + \
"\t" + repr( self.description ) + "\n" + \
"\t" + repr( self.trainings )
def __str__( self ):
return "Journal: \n" + \
"\t" + str( self.description ) + "\n" + \
"\t" + str( self.trainings )
def repr_for_json_dump( self ):
for_json = {}
trainings_for_json_dump = \
[ x.repr_for_json_dump() for x in self.trainings ]
for_json.update( self.description )
for_json.update( { 'trainings': trainings_for_json_dump } )
return for_json
def save_journal( self, filename ):
with open( filename, 'wt' ) as outfile:
json.dump( self.repr_for_json_dump(), outfile, indent = 2 )
@classmethod
def load_journal( cls, filename ):
journal = cls( [], {} )
with open( filename, 'rt' ) as infile:
dict_from_json = json.load( infile )
trainings = dict_from_json.pop('trainings')
for x in trainings:
journal.add_training( Training.init_from_json( x ) )
journal.add_description( dict_from_json )
return journal
def lookup_last_similar_exercise( self, exercise,
program_name = None,
last_training_index = None ):
if program_name is not None and \
last_training_index is not None:
for tr in reversed( self.trainings ):
if tr.description.get( 'training_program' ) == \
program_name \
and \
tr.description.get( 'training_index_in_program' ) == \
last_training_index:
for ex in reversed( tr.exercises ):
if ex.description.get('name') == \
exercise.description.get('name'):
return( ex )
for tr in reversed( self.trainings ):
for ex in reversed( tr.exercises ):
if ex.description.get('name') == \
exercise.description.get('name'):
return( ex )
return None
def get_dict_of_exercise_names_and_types( self ):
ex_names_and_types = {}
for tr in self.trainings:
for ex in tr.exercises:
ex_names_and_types.setdefault(
ex.description.get("name"),
ex.description.get("type") )
return( ex_names_and_types )