automatic language detection

quhacks · Dec 19, 2020 · 1c4b616 · 1c4b616
1 parent c120bd9
commit 1c4b616
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/main.py → judge.py b/main.py → judge.py
@@ -2,7 +2,8 @@
 
 file = sys.argv[1] if len(sys.argv) > 1 else input('Filename?')
 problem = input('Problem?')
-language = input('Language?').lower()
+extension = file.split('.')[-1] if '.' in file else ''
+language = extension if extension in ['py', 'java', 'class'] else input('Language?').lower()
 
 def get_args():
     if language in ['python', 'py', 'python3', 'py3']: