def setUp(self): app = Flask(__name__) app.config.from_object(TestSettings()) db.init_app(app) ctx = app.app_context() ctx.push() db.create_all() session = db.session() self.app = app self.db = db self.ctx = ctx self.session = session print "setUp"
def enc(s): if isinstance(s, unicode): return s.encode('UTF-8') else: return s app = Flask(__name__) app.config.from_object(IPythonSettings()) db.init_app(app) ctx = app.app_context() ctx.push() mysql = db.session() mg = MongoClient('192.168.5.202').zhihu print Question.query.delete() mysql.commit() #exit(1) cur = mg.questions.find() #.limit(100000) i = 0 for item in cur: try: d = item['data'] o = Question() o.title = enc(d['title']) o.id = int(d['id'])
def init(db): db.create_all() session = db.session() user1 = User() user1.id = 1 user1.username = '******' user1.password = '******' user1.name = 'shn7798_1' user2 = User() user2.id = 2 user2.username = '******' user2.password = '******' user2.name = 'shn7798_2' user3 = User() user3.id = 3 user3.username = '******' user3.password = '******' user3.name = 'shn7798_3' session.add_all([user1, user2, user3]) session.commit() u1 = User.query.filter(User.username == 'shn7798_1').first_or_404() u2 = User.query.filter(User.id == 2).first_or_404() assert (user1 == u1) assert (user2 == u2) # 创建问题1 question1 = Question() question1.id = 1 question1.title = 'question 1' question1.content = 'question 1 content' question1.user = user1 session.add(question1) session.commit() q1 = Question.query.filter(Question.id == 1).first_or_404() assert (q1.title == 'question 1') assert (q1.content == 'question 1 content') assert (q1.user_id == user1.id) # user2 创建答案 answer1 = Answer() answer1.id = 11 answer1.content = 'question 1 answer' answer1.user = user2 answer1.question = question1 session.add(answer1) session.commit() a1 = Answer.query.filter(Answer.id == 11).first_or_404() assert (a1.content == 'question 1 answer') assert (a1.user_id == user2.id) user1.op_on_questions.append(question1) uoq = user1.op_on_question(question1) uoq.follow = True session.commit() assert user1 in question1.following_users() op = user1.op_on_user(user2, edit=True) op.follow = True session.commit() user1.voteup_answer(answer1) session.commit() user1.block_user(user3) session.commit()
assert (a1.user_id == user2.id) user1.op_on_questions.append(question1) uoq = user1.op_on_question(question1) uoq.follow = True session.commit() assert user1 in question1.following_users() op = user1.op_on_user(user2, edit=True) op.follow = True session.commit() user1.voteup_answer(answer1) session.commit() user1.block_user(user3) session.commit() if __name__ == '__main__': app = create_app(IPythonSettings()) ctx = app.app_context() ctx.push() session = db.session()
from FlaskZhihu.extensions import db def enc(s): if isinstance(s, unicode): return s.encode('UTF-8') else: return s app = Flask(__name__) app.config.from_object(TestSettings()) db.init_app(app) ctx = app.app_context() ctx.push() mysql = db.session() mg = MongoClient('192.168.5.202').zhihu print Question.query.delete() mysql.commit() #exit(1) cur = mg.questions.find().limit(100000) i = 0 for item in cur: d = item['data'] o = Question() o.title = enc(d['title']) o.id = int(d['id']) o.excerpt = enc(d.get('excerpt', ''))