Update stopevents_gen.py

Updated the python code using the enumerate function, which will eliminate the need for an explicit index variable "i". It also uses f-strings for cleaner string formatting.
orioledb · Nov 9, 2023 · a6cb5cd · a6cb5cd
1 parent e6e4ac8
commit a6cb5cd
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/stopevents_gen.py b/stopevents_gen.py
@@ -12,17 +12,17 @@
 
 f = open('include/utils/stopevents_defs.h', 'w')
 f.write('/* Generated file, see stopevents_gen.py */\n\n')
-i = 0
-for e in event_names:
-	f.write('#define STOPEVENT_' + e.upper() + ' (' + str(i) + ')\n')
-	i = i + 1
-f.write('#define STOPEVENTS_COUNT (' + str(i) + ')\n')
+
+for i, e in enumerate(event_names):
+	f.write(f'#define STOPEVENT_{e.upper()} ({str(i)})\n')
+
+f.write('#define STOPEVENTS_COUNT (' + str(len(event_names)) + ')\n')
 f.close()
 
 f = open('include/utils/stopevents_data.h', 'w')
 f.write('/* Generated file, see stopevents_gen.py */\n\n')
-i = 0
+
 for e in event_names:
-	f.write('"' + e + '",\n')
-	i = i + 1
+	f.write(f'"{e}",\n')
+
 f.close()