demeter/demeter/db.py

# -*- coding: utf-8 -*-
"""
    demeter
    name:db.py
    author:rabin
"""

class Influxdb(object):
	"""
	instance = None
	def __new__(cls, *args, **kwd):
		if Influxdb.instance is None:
			Influxdb.instance = object.__new__(cls, *args, **kwd)
		return Influxdb.instance
	"""

	def __init__(self, config):
		influxdb = __import__('influxdb')
		InfluxDBClient = getattr(influxdb, 'InfluxDBClient')
		self.connect = InfluxDBClient(config['host'], config['port'], config['username'], config['password'], config['dbname'])
		self.create(config['dbname'])

	def get(self):
		return self.connect

	def create(self, name):
		database = self.connect.get_list_database()
		self.connect.create_database(name)


class Postgresql(object):
	"""
	instance = None
	def __new__(cls, *args, **kwd):
		if Postgresql.instance is None:
			Postgresql.instance = object.__new__(cls, *args, **kwd)
		return Postgresql.instance
	"""
	def __init__(self, config):
		psycopg2 = __import__('psycopg2')
		self.connect = psycopg2.connect(host=config['host'], port=config['port'], user=config['username'], password=config['password'], database=config['dbname'])

	def get(self):
		return self.connect

	def create(self, name):
		'psql -U postgres'
		sql = 'CREATE DATABASE '+name+' WITH OWNER = postgres ENCODING = "UTF8"'
		return sql

class Mysql(object):
	"""
	instance = None
	def __new__(cls, *args, **kwd):
		if Mysql.instance is None:
			Mysql.instance = object.__new__(cls, *args, **kwd)
		return Mysql.instance
	"""
	def __init__(self, config):
		pymysql = __import__('pymysql')
		self.connect = pymysql.connect(host=config['host'], port=int(config['port']), user=config['username'], password=config['password'], database=config['dbname'], charset=config['charset'])

	def get(self):
		return self.connect

	def create(self, name):
		sql = 'CREATE DATABASE IF NOT EXISTS '+name+' DEFAULT CHARSET utf8 COLLATE utf8_general_ci'
		return sql