声zzz

# coding: utf8
 
import _mysql
def quote(s):
  return _mysql.escape_string(s)
import hashlib
def hash(s):
  return hashlib.sha224(s).hexdigest()
import random
import time
import sys
 
_sql = {
  'db': """
    CREATE DATABASE %s 
    DEFAULT CHARACTER SET = utf8 
    DEFAULT COLLATE = utf8_general_ci
    """,
  'strings': """
    CREATE TABLE strings (
    name TEXT,
    hash CHAR(56),
    value TEXT,
    UNIQUE KEY (hash)
    ) ENGINE=MyISAM
    """,
  'lists': """
    CREATE TABLE lists (
    name TEXT,
    value TEXT,
    weight INT(64) DEFAULT 0
    ) ENGINE=MyISAM
    """,
  'sets': """
    CREATE TABLE sets (
    name TEXT,
    nhash CHAR(56),
    value TEXT,
    vhash CHAR(56),
    UNIQUE INDEX hash (nhash,vhash)
    ) ENGINE=MyISAM
    """,
}
 
class Mydis:
  def __init__(self, host = 'localhost', user = 'test', password = 'test', database = 'test'):
    self.init_args = (host, user, password, database)
    self.connection = _mysql.connect(host, user, password)
    try:
      self.connection.select_db(database)
    except Exception, e:
      if e[0] == 1049: #no database
        print 'creating database %s' % database
        self.connection.query(_sql['db'] % database)
        self.connection.select_db(database)
 
        self.connection.query(_sql['strings'])
        self.connection.query(_sql['lists'])
        self.connection.query(_sql['sets'])
      else:
        print e
        sys.exit()
 
    self.connection.set_character_set('UTF8')
    self.connection.query('SET NAMES UTF8')
 
  # strings
  def get(self, name):
    self._keep_alive()
    sql = "SELECT value FROM strings WHERE hash='%s' LIMIT 1" % hash(name)
    return self._get_one(sql)
 
  def set(self, name, value):
    self._keep_alive()
    if type(value) != str:
      value = str(value)
    sql = "INSERT INTO strings VALUES ('%s', '%s', '%s')" % (quote(name), hash(name), quote(value))
    try:
      self.connection.query(sql)
    except Exception, e:
      if e[0] == 1062:
        sql = "UPDATE strings SET value='%s' WHERE hash='%s'" % (quote(value), hash(name))
        self.connection.query(sql)
      else:
        print e
        sys.exit()
    return True
 
  def setnx(self, name, value):
    self._keep_alive()
    if type(value) != str:
      value = str(value)
    sql = "INSERT INTO strings VALUES ('%s', '%s', '%s')" % (quote(name), hash(name), quote(value))
    try:
      self.connection.query(sql)
    except Exception, e:
      if e[0] == 1062: pass
      else:
        print e
        sys.exit()
    return True
 
  def rem(self, name):
    self._keep_alive()
    sql = "DELETE FROM strings WHERE name='%s'" % quote(name)
    self.connection.query(sql)
    return True
 
  def keys(self, pattern):
    self._keep_alive()
    sql = "SELECT name FROM strings WHERE name LIKE '%s'" % quote(pattern)
    return self._get_col(sql)
 
  # lists
  def lpush(self, name, value):
    self._keep_alive()
    pass
 
  def rpush(self, name, value):
    self._keep_alive()
    pass
 
  def lpop(self, name):
    self._keep_alive()
    pass
 
  def rpop(self, name):
    self._keep_alive()
    pass
 
  def llen(self, name):
    self._keep_alive()
    pass
 
  def lrange(self, name, start, end):
    self._keep_alive()
    pass
 
  def lindex(self, name, index):
    self._keep_alive()
    pass
 
  def lset(self, name, index, value):
    self._keep_alive()
    pass
 
  # sets
  def sadd(self, name, value):
    self._keep_alive()
    if type(value) != str:
      value = str(value)
    sql = "INSERT INTO sets VALUES ('%s', '%s', '%s', '%s')" % (quote(name), hash(name), quote(value), hash(value))
    try:
      self.connection.query(sql)
    except Exception, e:
      if e[0] == 1062: pass
      else:
        print e
        sys.exit()
    return True
 
  def sget(self, name):
    self._keep_alive()
    sql = "SELECT value FROM sets WHERE nhash='%s'" % hash(name)
    return self._get_one(sql)
 
  def srem(self, name, value):
    self._keep_alive()
    if type(value) != str:
      value = str(value)
    sql = "DELETE FROM sets WHERE nhash='%s' AND vhash='%s'" % (hash(name), hash(value))
    self.connection.query(sql)
    return True
 
  def spop(self, name):
    self._keep_alive()
    tmp_id = hash(self._unique_id())
    sql = "UPDATE sets SET nhash='%s' WHERE nhash='%s' LIMIT 1" %  (tmp_id, hash(name))
    self.connection.query(sql)
    sql = "SELECT value FROM sets WHERE nhash='%s'" % tmp_id
    ret = self._get_one(sql)
    sql = "DELETE FROM sets WHERE nhash='%s'" % tmp_id
    self.connection.query(sql)
    return ret
 
  def skeys(self, pattern):
    self._keep_alive()
    sql = "SELECT DISTINCT name FROM sets WHERE name LIKE '%s'" % quote(pattern)
    return self._get_col(sql)
 
  def slen(self, name):
    self._keep_alive()
    sql = "SELECT COUNT(*) FROM sets WHERE nhash='%s'" % hash(name)
    return self._get_one(sql)
 
  def sismember(self, name, value):
    self._keep_alive()
    if type(value) != str:
      value = str(value)
    sql = "SELECT COUNT(*) FROM sets WHERE nhash='%s' AND vhash='%s'" % (hash(name), hash(value))
    return self._get_one(sql) == '1'
 
  def smembers(self, name):
    self._keep_alive()
    sql = "SELECT value FROM sets WHERE nhash='%s'" % hash(name)
    return self._get_col(sql)
 
  def sclear(self, name):
    self._keep_alive()
    sql = "DELETE FROM sets WHERE nhash='%s'" % hash(name)
    self.connection.query(sql)
    return True
 
  # misc
  def _keep_alive(self):
    try:
      self.connection.ping()
    except Exception, e:
      print 'reconnecting'
      if e[0] == 2006:
        self.__init__(*self.init_args)
      else:
        print e
        sys.exit()
 
  def _get_one(self, sql):
    self.connection.query(sql)
    res = self.connection.store_result()
    row = res.fetch_row()
    if row == ():
      return None
    return row[0][0]
 
  def _get_col(self, sql):
    self.connection.query(sql)
    res = self.connection.store_result()
    row = res.fetch_row()
    ret = []
    while row != ():
      ret.append(row[0][0])
      row = res.fetch_row()
    return ret
 
  def _unique_id(self):
    r = 0
    for i in xrange(64):
      r = r * 10 + random.randint(0, 9)
    return str(r)