# coding : utf8
import os
os.sys.path.append(os.getcwd())
from pymongo import MongoClient
from models import BookSource, Book
MONGO_DBS = ['xiaoshuo', 'xiaoshuo1', 'xiaoshuo2', 'xiaoshuo3', 'xiaoshuo4']
client = MongoClient('localhost', 27017)
source_site = 'hao123.se'
base_url = 'http://www.hao123.se/%s/'
all_books = Book.gets()
for book in all_books:
title = book.title
bid = book.id
for db in MONGO_DBS:
book_table = client[db].Book
book_in_mongo = book_table.find_one({"title": title})
if book_in_mongo:
source_url = base_url % book_in_mongo.get('bid')
BookSource.add(bid, source_site, source_url)
else:
continue
