operate_table.py

#!/usr/bin/env python3
#-*- coding:utf-8 -*-

'''
本文件用于创建和删除sqlite数据表
方便保存QQ动态数据

使用方法：
在shell下
创建表： python3 operate_table.py create_table
删除表： python3 operate_table.py drop_table
'''

import sqlite3
import sys


class Operate_table(object):

    def __init__(self):
        self.conn = sqlite3.connect('moods.sqlite')
        self.cur = self.conn.cursor()

    def create_table(self):
        sql = '''CREATE TABLE moods (
                 id integer primary key Autoincrement not null,
                 qq int not null,
                 content text null,
                 comment_count int not null,
                 ctime int not null,
                 phone text null,
                 image text null,
                 locate text null)'''
        self.cur.execute(sql)

    def drop_table(self):
        self.cur.execute('drop table moods')

if __name__ == '__main__':
    app = Operate_table()
    argv = sys.argv[1]
    
    if argv == 'create_table':
        app.create_table()
    elif argv == 'drop_table':
        app.drop_table()
    else:
        print("输入参数必须为create_table或drop_table其中之一")
        raise ValueError