Small #395 refactor

mymarilyn · Oct 5, 2023 · 0ce990b · 0ce990b
1 parent 6dd3ae5
commit 0ce990b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/clickhouse_driver/log.py b/clickhouse_driver/log.py
@@ -17,6 +17,8 @@
     'Test',
 )
 
+num_priorities = len(log_priorities)
+
 
 def log_block(block):
     if block is None:
@@ -27,7 +29,7 @@ def log_block(block):
     for row in block.get_rows():
         row = dict(zip(column_names, row))
 
-        if 1 <= row['priority'] <= 9:
+        if 1 <= row['priority'] <= num_priorities:
             priority = log_priorities[row['priority']]
         else:
             priority = row[0]