From 7855cfa5a2821f196d32b4613b8f2f8b15fcb2b9 Mon Sep 17 00:00:00 2001 From: apuayush Date: Mon, 9 Oct 2017 05:26:06 +0530 Subject: [PATCH 1/2] modified extension module --- Submission.py | 21 +++++++++++++-------- 1 file changed, 13 insertions(+), 8 deletions(-) diff --git a/Submission.py b/Submission.py index b813027..15de2c7 100644 --- a/Submission.py +++ b/Submission.py @@ -45,11 +45,16 @@ def fetchAndSave(self, i): def getExtension(self, lang): - if lang.find('JAVA') >= 0: - return '.java' - if lang.find('PYTH') >= 0: - return '.py' - if lang.find('C++') >= 0: - return '.cpp' - if lang.find('C') >= 0: - return '.c' + Ext = { + 'JAVA': '.java', + 'PYTH': '.py', + 'python3': '.py', + 'python2': '.py', + 'C++': '.cpp', + 'c++': '.cpp', + 'cpp': '.cpp', + 'c': '.c', + 'C': '.c' + } + + return Ext[lang] From 9bc8517be8eaa9b2f831886f6270c5bc6fd1386c Mon Sep 17 00:00:00 2001 From: apuayush Date: Mon, 9 Oct 2017 06:39:02 +0530 Subject: [PATCH 2/2] Fixes #2 .added more languages --- Submission.py | 23 ++++++++++++++++++----- 1 file changed, 18 insertions(+), 5 deletions(-) diff --git a/Submission.py b/Submission.py index 15de2c7..971491f 100644 --- a/Submission.py +++ b/Submission.py @@ -46,15 +46,28 @@ def fetchAndSave(self, i): def getExtension(self, lang): Ext = { - 'JAVA': '.java', - 'PYTH': '.py', + 'java': '.java', + 'python': '.py' + 'pyth': '.py', 'python3': '.py', 'python2': '.py', - 'C++': '.cpp', 'c++': '.cpp', 'cpp': '.cpp', 'c': '.c', - 'C': '.c' + "pascal": ".pas", + "ruby": ".rb", + "go": ".go", + "javascript": ".js", + "shell": ".sh", + "r":".r", + 'c#': '.cs', + 'php': '.php', + 'bash': '.bash', + 'haskell': '.hs', + 'ruby': '.rb', + 'perl': '.pl', + 'scala': '.scala', + 'text': '.txt' } - return Ext[lang] + return Ext[lang.lower()]