def init_blog_db(): blog1 = Blog(title="954.二倍数对数组") blog1.url_id = gen_blog_url_id() cat1 = BlogCategory(name="LeetCode") cat2 = BlogCategory(name="Python3") cat3 = BlogCategory(name="AI&ML") cat4 = BlogCategory(name="C++") blog1.category = cat1 db.session.add_all([blog1]) db.session.add_all([cat1, cat2, cat3, cat4]) db.session.commit()
def create_blog(cat_id): if request.method == 'GET': cat_all = BlogCategory.query.filter().all() blog_cat = BlogCategory.query.filter(BlogCategory.id == int(cat_id)).first() dicts = { 'cat_all': cat_all, 'note_cat': blog_cat, 'catid': cat_id } return render_template('blog/create_blog.html', **dicts) else: title = request.form.get('docTitle') if title: new_blog = Blog(title=title) new_blog.url_id = gen_blog_url_id() username = session.get('username') # 指定用户 if username: user = User.query.filter(User.username == username).first() else: user = User.query.filter().first() new_blog.user = user # 设定类别 note_cat = BlogCategory.query.filter(BlogCategory.id == int(cat_id)).first() new_blog.category = note_cat # 记录内容 content = request.form.get('docContent') path = FILESERVER + PATHSEP + 'blogs' + PATHSEP + cat_id + secure_filename(''.join(lazy_pinyin(note_cat.name))) + PATHSEP full_filename = path + secure_filename(''.join(lazy_pinyin(title))) + '.md' if not os.path.exists(path): os.makedirs(path) sec_writefile(full_filename, content) new_blog.content_path = full_filename # 更新数据库 db.session.add(new_blog) db.session.commit() return redirect(url_for('main.blogs_cat', note_catid=int(cat_id)))