全宋词爬虫数据模型脚本

crawl_songci_db.py

# -*- coding: utf-8 -*-
'''
File Name: db.py
Author: JackeyGao
mail: junqi.gao@shuyun.com
'''

from peewee import *

db = SqliteDatabase('ci.db')

class CiAuthor(Model):
    value = IntegerField(primary_key=True)
    name = CharField()
    long_desc = TextField(null=True)
    short_desc = TextField(null=True)

    class Meta:
        database = db # This model uses the "people.db" database.

class Ci(Model):
    value = IntegerField(primary_key=True)
    rhythmic = CharField()
    author = CharField()
    content = TextField(null=True)

    class Meta:
        database = db


#def delete_note():
#    return Note.delete().execute()
#
#def delete_image():
#    return Image.delete().execute()

def init_table():
    db.connect()
    db.create_tables([Ci, CiAuthor])

if __name__ == '__main__':
    init_table()