#!/usr/bin/env python
# -*- coding: utf-8 -*-
import sqlite3
import logging
import logging.config
logging.config.fileConfig('logging.conf')
levels = {
    1: 'CRITICAL',
    2: 'ERROR',
    3: 'WARNING',
    4: 'INFO',
    5: 'DEBUG',
}
loglevel = 4
logger = logging.getLogger('spider')
logger.setLevel(levels[loglevel])
class MySqlite(object):
    def __init__(self, dbfile):
        try:
            self.conn = sqlite3.connect(dbfile)
            logger.info("Open database %s" % dbfile)
            logger.debug("Open database %s" % dbfile)
        except sqlite3.Error as e:
            # print "Fail to connect %s: %s" % (dbfile, e) # e.args[0]
            logger.error("Fail to connect %s: %s" % (dbfile, e))
            return
        self.cursor = self.conn.cursor()
    def create(self, table):
        try:
            logger.info("Create table %s" % table)
            self.cursor.execute(
                "CREATE TABLE IF NOT EXISTS %s(Id INTEGER PRIMARY KEY \
                AUTOINCREMENT, Data VARCHAR(40))" % table
                )
            self.conn.commit()
        except sqlite3.Error as e:
            logger.error("Fail to create %s: %s" % (table, e))
            self.conn.rollback()
    def insert(self, table, data):
        try:
            logger.info("Insert %s into table %s" % (data, table))
            self.cursor.execute(
                "INSERT INTO %s(Data) VALUES('%s')" % (table, data))
            self.conn.commit()
        except sqlite3.Error as e:
            logger.error("Fail to insert %s into %s: %s" % (data, table, e))
            self.conn.rollback()
    def close(self):
        logger.info("Close database")
        self.cursor.close()
        self.conn.close()
if __name__ == '__main__':
    ms = MySqlite('spider.db')
    ms.create('t1')
    ms.insert('t1', 'test')
    ms.close()