fix : RANDOM_FILTERED_ARTS added #39

sepandhaghighi · Apr 6, 2019 · 24e87c3 · 24e87c3
1 parent 9172004
commit 24e87c3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/art/art.py b/art/art.py
@@ -141,8 +141,9 @@ def art(artname, number=1, text=""):
         raise artError(ART_TYPE_ERROR)
     artname = artname.lower()
     arts = sorted(art_dic.keys())
-    if artname == "random" or artname == "rand":
-        artname = random.choice(arts)
+    if artname == "random" or artname == "rand" or artname == "rnd":
+        filtered_arts = list(set(arts) - set(RANDOM_FILTERED_ARTS))
+        artname = random.choice(filtered_arts)
     elif artname not in art_dic.keys():
         distance_list = list(map(lambda x: distance_calc(artname, x),
                                  arts))

diff --git a/art/art_param.py b/art/art_param.py
@@ -150,6 +150,10 @@
     "mirror_flip",
     "flip"]
 
+RANDOM_FILTERED_ARTS = ["message2","love you","text decoration","message1","musical"]
+
+
+
 TEST_FILTERED_FONTS = ["mirror", "mirror_flip", "flip"]
 
 DESCRIPTION = '''ASCII art is also known as "computer text art".