manage.py

# -*- encoding=UTF-8 -*-

from nowstagram import app, db
from flask_script import Manager
from sqlalchemy import or_,and_
from nowstagram.models import User, Image, Comment
import random,unittest
import requests
import json
import urllib
manager = Manager(app)

imgs_url=[]
def get_image_url():
    global imgs_url
    imgs = requests.get('http://pic.sogou.com/pics/channel/getAllRecomPicByTag.jsp?category=壁纸'+'&tag=%E5%85%A8%E9%83%A8&start=0&len=2000')
    jd = json.loads(imgs.text)
    jd=jd['all_items']
    imgs_url=[]
    for j in jd:
        imgs_url.append(j['bthumbUrl'])
    return None

@manager.command
def init_database():
    db.drop_all()
    db.create_all()
    get_image_url()
    for i in range(0, 100):
        db.session.add(User('User' + str(i+1), 'a'+str(i+1)))
        for j in range(0, 10):
            db.session.add(Image(imgs_url[random.randint(0,j*200)], i+1))
            for k in range(0, 3):
                db.session.add(Comment('This is a comment' + str(k), 1+10*i+j, i+1))

    db.session.commit()

@manager.command
def run_test():
    db.drop_all()
    db.create_all()
    tests = unittest.TestLoader().discover('./')
    unittest.TextTestRunner().run(tests)

   # for i in range(50, 100, 2):
     #   user = User.query.get(i)
    #    user.username = '[New1]' + user.username

   # User.query.filter_by(id=51).update({'username':'[New2]'})
   # db.session.commit()

   # for i in range(50, 100, 2):
    #    comment = Comment.query.get(i+1)
   #     db.session.delete(comment)
  #  db.session.commit()

   # print 1, User.query.all()
   # print 2, User.query.get(3)
   # print 3, User.query.filter_by(id=5).first()
  #  print 4, User.query.order_by(User.id.desc()).offset(1).limit(2).all()
   # print 5, User.query.filter(User.username.endswith('0')).limit(3).all()
    #print 6, User.query.filter(or_(User.id == 88, User.id == 99)).all()
   # print 7, User.query.filter(and_(User.id > 88, User.id < 93)).all()
   # print 8, User.query.filter(and_(User.id > 88, User.id < 93)).first_or_404()
   # print 9, User.query.order_by(User.id.desc()).paginate(page=1, per_page=10).items
   # user = User.query.get(1)
   # print 10, user.images



   # image = Image.query.get(1)
   # print 11, image, image.user

if __name__ == '__main__':
    manager.run()