[s7] change match arg

s7.py

@@ -12,13 +12,15 @@ import random
 '''
 แก้ตรงนี้ไปก่อนละกัน No of Tree = “4Ts” และ No of R = “5”
 
-No of Tree ของคุณ ==== NOT_ARG --- ถ้าต้องการทั้งหมดใช้ 'all'
+No of Tree ของคุณ ==== NoT_MATCH_ARG เป็น list หรือ tuple
+    - ถ้าจะเอาอันไหนก็ใส่ใน list ไป เช่น เอา 4Ts, 2Ts >> ['4Ts', '2Ts']
+    - ถ้าเอาทุกอันจะใส่ ['4Ts', '3Ts', ... ] ให้ครบ หรือใช้ [] ก็ได้
 No of R ของคุณ ======= TOTAL -- ถ้าต้องการทั้งหมดก็ = 0
 แต่ว่า ผมไม่รับรองว่าจะใช้ได้ทั้งหมดเพราะว่า มันก็ขึ้นกับหลายอย่าง
 เช่น 1Ts มีไม่กี่อันอาจจะ error มั้ง? โดยเฉพาะเวลา random
 '''
-NOT_ARG = 'all'
+NoT_MATCH_ARG = ['2Ts', '1Ts']
-TOTAL = 20
+TOTAL = 10
 
 SRC_PATH = os.path.join(TARGET_DIR, 'part6')
 
@@ -40,7 +42,7 @@ def get_by_order():
         with open(os.path.join(SRC_PATH, '%s-6.csv' % m), 'rb') as f:
             rows = csv.reader(f)
             for r in rows:
-                if NOT_ARG != r[2] and NOT_ARG != 'all':
+                if NoT_MATCH_ARG and r[2] not in NoT_MATCH_ARG:
                     continue
                 row = [
                     r[0],  # R_id
@@ -81,7 +83,7 @@ def get_by_random():
             ok = False
             while not ok:
                 pick = random.choice(lo_list)
-                if NOT_ARG != pick[2] and NOT_ARG != 'all':
+                if NoT_MATCH_ARG and pick[2] not in NoT_MATCH_ARG:
                     continue
                 if pick[0] not in picked:
                     picked.append(pick[0])