forked from tanner/qotnews
46 lines
1.0 KiB
Python
46 lines
1.0 KiB
Python
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
|
|
|