add a solution

howel52 · howel52 · commit 7dc56e75319c · 2016-06-03T13:07:17.000+08:00
diff --git a/lenzzz/0002/main.py b/lenzzz/0002/main.py
@@ -0,0 +1,32 @@
+from random import randint
+import MySQLdb
+def makeCode(length,number):
+    code = []
+    code_set = set(code)
+    code_map = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+    limit = len(code_map)
+    while(len(code_set)<number):
+        this_code = ""
+        for i in range(length):
+            this_code += code_map[randint(0,limit-1)]
+        code.append(this_code)
+        code_set = set(code)
+
+    return code_set
+
+def insertDatebase(code):
+    db = MySQLdb.Connect("localhost","root","","pythonmysql")
+    cursor = db.cursor()
+    sql = """INSERT INTO my_practice
+         VALUES ('%s')""" % code
+    try:
+        cursor.execute(sql)
+        db.commit()
+    except:
+        db.rollback()
+    db.close()
+
+if __name__ == '__main__':
+    codes = makeCode(20,200)
+    for code in codes:
+        insertDatebase(code)
