parent
7a131ebd03
commit
dcedd4caa1
3 changed files with 67 additions and 56 deletions
@ -0,0 +1,45 @@ |
||||
import logging |
||||
logging.basicConfig( |
||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', |
||||
level=logging.INFO) |
||||
|
||||
import database |
||||
from sqlalchemy import select |
||||
import search |
||||
import sys |
||||
|
||||
import json |
||||
import requests |
||||
|
||||
database.init() |
||||
search.init() |
||||
|
||||
def count_stories(): |
||||
try: |
||||
session = database.Session() |
||||
return session.query(database.Story).count() |
||||
finally: |
||||
session.close() |
||||
|
||||
def get_story_list(): |
||||
try: |
||||
session = database.Session() |
||||
return session.query(database.Story.sid).all() |
||||
finally: |
||||
session.close() |
||||
|
||||
if __name__ == '__main__': |
||||
num_stories = count_stories() |
||||
|
||||
print('Reindex {} stories?'.format(num_stories)) |
||||
print('Press ENTER to continue, ctrl-c to cancel') |
||||
input() |
||||
|
||||
count = 1 |
||||
for sid in get_story_list(): |
||||
story = database.get_story(sid) |
||||
print('Indexing {}/{} id: {} title: {}'.format(count, num_stories, sid[0], story.title)) |
||||
story_obj = json.loads(story.meta_json) |
||||
search.put_story(story_obj) |
||||
count += 1 |
||||
|
Loading…
Reference in new issue