29 lines
818 B
Python
29 lines
818 B
Python
from sqlalchemy import create_engine, Column, Integer, String
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
import os
|
|
|
|
engine = create_engine('sqlite:///douban.db')
|
|
Base = declarative_base(engine)
|
|
|
|
|
|
class Douban(Base):
|
|
__tablename__ = 'DouBan'
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
tag = Column(String, unique=False, nullable=False)
|
|
url = Column(String, unique=True, nullable=False)
|
|
|
|
def __repr__(self):
|
|
return "<Douban(id='%d', tag='%s',url='%s')>" % (self.id, self.tag, self.url)
|
|
|
|
|
|
if os.path.isfile('douban.db') is False:
|
|
print('正在创建数据库...')
|
|
Base.metadata.create_all()
|
|
else:
|
|
print('检测到现有数据库，正在读取...')
|
|
|
|
if __name__ == '__main__':
|
|
# 重置数据库
|
|
Base.metadata.drop_all(engine)
|
|
print('Done')
|