def add_title(site, count=100):
print("titles")
# TODO: Make sure that every slug is unique
chain = get_chain(site.url, "Titles")
for _ in range(count):
title = generate_text(chain, "Titles")
slug = slugify(title, max_length=70, word_boundary=True)
Title.create(text=title, slug=slug, site=site)
def add_answer(site, count=300):
print("answers")
users = get_unused_users(site, count)
chain = get_chain(site.url, "Answers")
for i in range(count):
text = generate_text(chain, "Answers")
user = users[i]
time = datetime.now()
Answer.create(text=text, user_id=user, site_id=site, datetime=time)
def worker_voice(event_voice):
"""音声再生のためのworker
"""
while not stop:
# event.set が実行されるまで待機
event_voice.wait()
event_voice.clear()
logging.debug('voice start')
text = text_generator.generate_text(working_state)
# time.sleep(3)
sound.play_sound(text)
logging.debug('voice end')
def worker_notification(event_notification):
"""通知のためのworker
"""
global working_state
while not stop:
# event.set が実行されるまで待機
event_notification.wait()
event_notification.clear()
logging.debug('notify start')
text = text_generator.generate_text(working_state)
img_path = notification.generate_path(working_state)
notification.notify(text, img_path)
logging.debug('notify end')
if working_state == "praise":
title = activity_log.get_active_window_title()
activity_log.set_start_word_num(title.split()[0])
working_state = "idle"
def initialize_output():
""" Create/clear the output directory """
OUT_DIR = 'output'
if not os.path.exists(OUT_DIR):
os.makedirs(OUT_DIR)
junk = [x for x in os.listdir(OUT_DIR) if x[-4:] == '.jpg']
for file in junk:
os.unlink(OUT_DIR + '/' + file)
print('Output directory initialized.')
print('')
for i in range(MAX_INSTANCES):
top_line, bottom_line = generate_text()
image_name = get_random_input_file()
image = write_text_to_image(top_line, bottom_line, image_name)
print('Writing to {}:'.format(image_name))
print('\t{}'.format(top_line))
print('\t{}'.format(bottom_line))
save_image(image, image_name)
def add_question(site, count=100):
print("Questions")
users = get_unused_users(site, count)
titles = Title.select().join(Question, JOIN.LEFT_OUTER) \
.where((Title.site == site) & (Question.id.is_null())) \
.limit(count)
chain = get_chain(site.url, "Questions")
for i in range(count):
text = generate_text(chain, "Questions")
title = titles[i]
user = users[i]
time = datetime.now()
question = Question.create(text=text,
title_id=title,
user_id=user,
site_id=site,
datetime=time)
num_answers = random.randint(1, 4)
answers = Answer.select().where((Answer.site == site) & (
Answer.question.is_null())).limit(num_answers)
for answer in answers:
answer.question = question
answer.save()
def add_username(site, count=500):
print("usernames")
chain = get_chain(site.url, "Usernames")
for _ in range(count):
username = generate_text(chain, "Usernames")
User.create(username=username, site=site)