change ProgLang from class to namedtuple

bonfy · bonfy · commit c11d04d71368 · 2016-12-02T20:23:21.000+08:00
diff --git a/leetcode_generate.py b/leetcode_generate.py
@@ -15,6 +15,7 @@
 from pyquery import PyQuery as pq
 import re
 import sys
+from collections import namedtuple
 
 
 HOME = os.getcwd()
@@ -79,11 +80,7 @@ def check_and_make_dir(dirname):
         os.mkdir(dirname)
 
 
-class ProgLang:
-    def __init__(self, language, ext, annotation):
-        self.language = language
-        self.ext = ext
-        self.annotation = annotation
+ProgLang = namedtuple('ProgLang', ['language', 'ext', 'annotation'])
 
 ProgLangList = [ProgLang('c++', 'cpp', '//'),
                 ProgLang('java', 'java', '//'),
