From 4f5909ad219c68f9500f324f5cfe7af04fc01209 Mon Sep 17 00:00:00 2001 From: Tiago Date: Wed, 11 Aug 2021 18:42:03 -0300 Subject: [PATCH] Generate a small random size when necessary --- awxkit/awxkit/utils/__init__.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/awxkit/awxkit/utils/__init__.py b/awxkit/awxkit/utils/__init__.py index a3b1fa6df4..c3f9e1bbe2 100644 --- a/awxkit/awxkit/utils/__init__.py +++ b/awxkit/awxkit/utils/__init__.py @@ -276,13 +276,14 @@ def random_utf8(*args, **kwargs): def random_title(num_words=2, non_ascii=True): - if os.getenv('AWXKIT_FORCE_ONLY_ASCII', False): - non_ascii = False base = ''.join([random.choice(words) for word in range(num_words)]) - if non_ascii: - title = ''.join([base, random_utf8(1)]) + if os.getenv('AWXKIT_FORCE_ONLY_ASCII', False): + title = ''.join([base, ''.join(str(random_int(99)))]) else: - title = ''.join([base, ''.join([str(random_int()) for _ in range(3)])]) + if non_ascii: + title = ''.join([base, random_utf8(1)]) + else: + title = ''.join([base, ''.join([str(random_int()) for _ in range(3)])]) return title