diff --git a/logic.py b/logic.py index 30eae82..821ea3e 100644 --- a/logic.py +++ b/logic.py @@ -118,8 +118,8 @@ async def secret_cache_key(self): self._secret_cache_key = self._secret_cache_key_fmt.format(self.secret, self.date) return self._secret_cache_key - def _get_secret_vector(self): - return self.vector_logic.get_vector(self.secret) + async def _get_secret_vector(self): + return await self.vector_logic.get_vector(self.secret) def _iterate_all_wv(self): return self.vector_logic.iterate_all() @@ -133,7 +133,7 @@ async def set_secret(self, dry=False, force=False): if wv.get('secret_date') is not None: raise ValueError("This word was a secret in the past") - secret_vec = self._get_secret_vector() + secret_vec = await self._get_secret_vector() nearest = [] async for word, vec in self._iterate_all_wv(): diff --git a/scripts/set_secret.py b/scripts/set_secret.py index 951ee95..bc9327a 100644 --- a/scripts/set_secret.py +++ b/scripts/set_secret.py @@ -81,7 +81,7 @@ async def do_populate(mongo, redis, model, secret, date, force): else: logic = CacheSecretLogic(mongo, redis, secret, date) await logic.set_secret(dry=True, force=force) - cache = [w[::-1] for w in await logic.cache[::-1]] + cache = [w[::-1] for w in (await logic.cache)[::-1]] print(' ,'.join(cache)) print(cache[0]) for rng in (range(i, i+10) for i in [1, 50, 100, 300, 550, 750]): @@ -95,7 +95,7 @@ async def do_populate(mongo, redis, model, secret, date, force): return True else: secret = await get_random_word(mongo) - return do_populate(mongo, redis, model, secret, date, force) + return await do_populate(mongo, redis, model, secret, date, force) async def get_random_word(mongo):