diff --git a/rep_cli.py b/rep_cli.py index 14d0ea4..aa39826 100644 --- a/rep_cli.py +++ b/rep_cli.py @@ -46,7 +46,24 @@ class Session: def result(self): return self.resultstring + + def movement_is_mastered(self, work_id, mov_id): + is_mastered = False + if self.user != -1: + sql_command = f''' + SELECT * + FROM is_able_to_play + WHERE work_id = {work_id} + AND mov_id = {mov_id} + AND pianist_id = {self.user} + ''' + if self.db_agent.execute(sql_command).fetchone(): + is_mastered = True + return is_mastered + def work_is_mastered(self, work): + pass + def create_new_user(self, arguments): first_name, name = arguments[0], arguments[1] sql_command = f''' @@ -115,6 +132,10 @@ class Session: work = work_under_id(work_id, self.db_agent) fun_resultstring = '' for mov_number in work.values['movements'].keys(): + if self.movement_is_mastered(work_id, mov_number): + fun_resultstring += '[x] ' + else: + fun_resultstring += '[ ] ' fun_resultstring += f'{mov_number}. {work.pretty_mov(mov_number)}\n' return fun_resultstring