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 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', ''))