Fix minor bug

Author: RussellDash332

autokattis/api/databasemanager.py

@@ -11,7 +11,7 @@ def __init__(self, user):
             name = option.text.strip()
             if val and name: self.LANGUAGES[val] = name
         for val, name in [*self.LANGUAGES.items()]:
-            self.LANGUAGES[name] = val
+            self.LANGUAGES[name] = self.LANGUAGES[val] = val
         print('[database] Listed all available languages!', flush=True)
 
         self.COUNTRIES = {}

autokattis/api/nuskattis.py

@@ -354,7 +354,7 @@ def stats(self, languages=None, *bc_args):
             params = {
                 'page': 0,
                 'status': 'AC',
-                'language': language
+                'language': self.get_database().get_languages().get(language)
             }
             data = {}
             with ThreadPoolExecutor(max_workers=self.get_max_workers()) as executor:
@@ -467,7 +467,7 @@ def offerings(self, course_id):
                 name, end_date = [truncate_spaces(column.text.strip()) for column in columns]
                 link, _ = [column.find('a') for column in columns]
                 data.append({
-                    'name': name,
+                    'name': name.replace('\n', ''),
                     'end_date': end_date.split()[1][:-1],
                     'link': self.get_base_url() + link.get('href')
                 })

autokattis/api/openkattis.py

@@ -211,11 +211,12 @@ def plot_problems(self, filepath=None, show_solved=True, show_partial=True, show
         categories = set(df.category)
 
         hue_order = [enum_to_title(c) for c in [DifficultyColor.N_A, DifficultyColor.HARD, DifficultyColor.MEDIUM, DifficultyColor.EASY] if enum_to_title(c) in categories]
-        palette = {enum_to_title(c):c.value for c in [DifficultyColor.EASY, DifficultyColor.MEDIUM, DifficultyColor.HARD, DifficultyColor.NA] if c in categories}
+        palette = {enum_to_title(c):c.value for c in [DifficultyColor.EASY, DifficultyColor.MEDIUM, DifficultyColor.HARD, DifficultyColor.N_A] if enum_to_title(c) in categories}
 
         diff_lo = math.floor(min([d for d in df.difficulty if d], default=0))
         diff_hi = math.ceil(max([d for d in df.difficulty if d], default=0))
 
+        plt.clf()
         hist = sns.histplot(data=df, x='difficulty', hue='category', multiple='stack', bins=[i/10 for i in range(10*diff_lo, 10*diff_hi+1)], hue_order=hue_order, palette=palette)
         hist.set(title=f'Solved Kattis Problems by {self.get_username()} ({df.shape[0]})', xlabel='Difficulty')
         plt.legend(title='Category', loc='upper right', labels=hue_order[::-1])
@@ -505,7 +506,7 @@ def stats(self, languages=None, *bc_args):
             params = {
                 'page': 0,
                 'status': 'AC',
-                'language': language
+                'language': self.get_database().get_languages().get(language)
             }
             data = {}
             with ThreadPoolExecutor(max_workers=self.get_max_workers()) as executor:

test/nuskattis.py

@@ -20,9 +20,9 @@
 
 test('nus_courses',                         kt.courses,         {})
 
-test('nus_offerings_valid_1',               kt.courses,         {'course_id': 'CS2040'})
-test('nus_offerings_valid_2',               kt.courses,         {'course_id': 'CS3233'})
-test('nus_offerings_invalid',               kt.courses,         {'course_id': 'idk'})
+test('nus_offerings_valid_1',               kt.offerings,       {'course_id': 'CS2040'})
+test('nus_offerings_valid_2',               kt.offerings,       {'course_id': 'CS3233'})
+test('nus_offerings_invalid',               kt.offerings,       {'course_id': 'idk'})
 
-test('nus_assignments_courseid_guessed',    kt.assignments,     {'offering_id': 'CS3233_S2_AY2223'})
+test('nus_assignments_courseid_guessed',    kt.assignments,     {'offering_id': 'CS2040_S1_AY2425'})
 test('nus_assignments_courseid_given',      kt.assignments,     {'offering_id': 'CS3233_S2_AY2223', 'course_id': 'CS3233'})

test/openkattis.py

@@ -5,10 +5,10 @@
 kt = Kattis(USER, PASSWORD)
 kt = OpenKattis(USER, PASSWORD)
 
-test('open_problems_solved_ldm',            kt.problems,            {})
-test('open_problems_solved_fdm',            kt.problems,            {'low_detail_mode': False})
-test('open_problems_all_ldm',               kt.problems,            {'show_tried': True, 'show_untried': True})
-test('open_problems_all_fdm',               kt.problems,            {'show_tried': True, 'show_untried': True, 'low_detail_mode': False})
+test('open_problems_solved_fdm',            kt.problems,            {})
+test('open_problems_solved_ldm',            kt.problems,            {'low_detail_mode': True})
+test('open_problems_all_fdm',               kt.problems,            {'show_tried': True, 'show_untried': True})
+test('open_problems_all_ldm',               kt.problems,            {'show_tried': True, 'show_untried': True, 'low_detail_mode': True})
 
 test('open_plot_problems_solved_nofile',    kt.plot_problems,       {})
 test('open_plot_problems_solved_file',      kt.plot_problems,       {'filepath': 'plot1.png'})

test/utils.py

@@ -20,11 +20,12 @@ def test(title, fn, kwargs, csv=True):
     start = time.perf_counter()
     try:
         ret = fn(**kwargs)
-        print(df:=ret.to_df())
-        if csv: df.to_csv(f'{title}.csv', index=False)
+        if ret != None:
+            print(df:=ret.to_df())
+            if csv: df.to_csv(f'{title}.csv', index=False)
         print(f'Time taken: {round(time.perf_counter()-start, 4)}s')
         print(f'[{bcolors.OKGREEN}OK{bcolors.ENDC}] {fn.__qualname__}:{title}')
     except Exception as e:
         print(f'{type(e).__name__}: {e}')
         print(f'Time taken: {round(time.perf_counter()-start, 4)}s')
-        print(f'[{bcolors.FAIL}FAIL{bcolors.ENDC}] {fn.__qualname__}:{title}')
+        print(f'[{bcolors.FAIL}FAIL{bcolors.ENDC}] {fn.__qualname__}:{title}')