#!/usr/bin/env python
# -*- coding: utf-8 -*-
from functools import partial
from pymysql.converters import escape_item, escape_string
import sysPY2 = sys.version_info[0] == 2if PY2:import __builtin__range_type = xrangetext_type = unicodelong_type = longstr_type = basestringunichr = __builtin__.unichr
else:range_type = rangetext_type = strlong_type = intstr_type = strunichr = chrdef _ensure_bytes(x, encoding="utf8"):if isinstance(x, text_type):x = x.encode(encoding)return xdef _escape_args(args, encoding):ensure_bytes = partial(_ensure_bytes, encoding=encoding)if isinstance(args, (tuple, list)):if PY2:args = tuple(map(ensure_bytes, args))return tuple(escape(arg, encoding) for arg in args)elif isinstance(args, dict):if PY2:args = dict((ensure_bytes(key), ensure_bytes(val)) for(key, val) in args.items())return dict((key, escape(val, encoding)) for (key, val) in args.items())def escape(obj, charset, mapping=None):if isinstance(obj, str_type):return "'" + escape_string(obj) + "'"return escape_item(obj, charset, mapping=mapping)def mogrify(query, encoding, args=None):if PY2:  # Use bytes on Python 2 alwaysquery = _ensure_bytes(query, encoding=encoding)if args is not None:# r = _escape_args(args, encoding)query = query % _escape_args(args, encoding)return query# 调用一下mogrigy函数
# print(mogrify("select * from ll where a in %s and b = %s", "utf8", [[2, 1], 3]))

def _convert_second_fraction(s):if not s:return 0# Pad zeros to ensure the fraction length in microsecondss = s.ljust(6, '0')return int(s[:6])DATETIME_RE = re.compile(r"(\d{1,4})-(\d{1,2})-(\d{1,2})[T ](\d{1,2}):(\d{1,2}):(\d{1,2})(?:.(\d{1,6}))?")def convert_datetime(obj):"""Returns a DATETIME or TIMESTAMP column value as a datetime object:>>> datetime_or_None('2007-02-25 23:06:20')datetime.datetime(2007, 2, 25, 23, 6, 20)>>> datetime_or_None('2007-02-25T23:06:20')datetime.datetime(2007, 2, 25, 23, 6, 20)Illegal values are returned as None:>>> datetime_or_None('2007-02-31T23:06:20') is NoneTrue>>> datetime_or_None('0000-00-00 00:00:00') is NoneTrue"""if not PY2 and isinstance(obj, (bytes, bytearray)):obj = obj.decode('ascii')m = DATETIME_RE.match(obj)if not m:return convert_date(obj)try:groups = list(m.groups())groups[-1] = _convert_second_fraction(groups[-1])return datetime.datetime(*[ int(x) for x in groups ])except ValueError:return convert_date(obj)TIMEDELTA_RE = re.compile(r"(-)?(\d{1,3}):(\d{1,2}):(\d{1,2})(?:.(\d{1,6}))?")def convert_timedelta(obj):"""Returns a TIME column as a timedelta object:>>> timedelta_or_None('25:06:17')datetime.timedelta(1, 3977)>>> timedelta_or_None('-25:06:17')datetime.timedelta(-2, 83177)Illegal values are returned as None:>>> timedelta_or_None('random crap') is NoneTrueNote that MySQL always returns TIME columns as (+|-)HH:MM:SS, butcan accept values as (+|-)DD HH:MM:SS. The latter format will notbe parsed correctly by this function."""if not PY2 and isinstance(obj, (bytes, bytearray)):obj = obj.decode('ascii')m = TIMEDELTA_RE.match(obj)if not m:return Nonetry:groups = list(m.groups())groups[-1] = _convert_second_fraction(groups[-1])negate = -1 if groups[0] else 1hours, minutes, seconds, microseconds = groups[1:]tdelta = datetime.timedelta(hours = int(hours),minutes = int(minutes),seconds = int(seconds),microseconds = int(microseconds)) * negatereturn tdeltaexcept ValueError:return NoneTIME_RE = re.compile(r"(\d{1,2}):(\d{1,2}):(\d{1,2})(?:.(\d{1,6}))?")def convert_time(obj):"""Returns a TIME column as a time object:>>> time_or_None('15:06:17')datetime.time(15, 6, 17)Illegal values are returned as None:>>> time_or_None('-25:06:17') is NoneTrue>>> time_or_None('random crap') is NoneTrueNote that MySQL always returns TIME columns as (+|-)HH:MM:SS, butcan accept values as (+|-)DD HH:MM:SS. The latter format will notbe parsed correctly by this function.Also note that MySQL's TIME column corresponds more closely toPython's timedelta and not time. However if you want TIME columnsto be treated as time-of-day and not a time offset, then you canuse set this function as the converter for FIELD_TYPE.TIME."""if not PY2 and isinstance(obj, (bytes, bytearray)):obj = obj.decode('ascii')m = TIME_RE.match(obj)if not m:return Nonetry:groups = list(m.groups())groups[-1] = _convert_second_fraction(groups[-1])hours, minutes, seconds, microseconds = groupsreturn datetime.time(hour=int(hours), minute=int(minutes),second=int(seconds), microsecond=int(microseconds))except ValueError:return Nonedef convert_date(obj):"""Returns a DATE column as a date object:>>> date_or_None('2007-02-26')datetime.date(2007, 2, 26)Illegal values are returned as None:>>> date_or_None('2007-02-31') is NoneTrue>>> date_or_None('0000-00-00') is NoneTrue"""if not PY2 and isinstance(obj, (bytes, bytearray)):obj = obj.decode('ascii')try:return datetime.date(*[ int(x) for x in obj.split('-', 2) ])except ValueError:return Nonedef convert_mysql_timestamp(timestamp):"""Convert a MySQL TIMESTAMP to a Timestamp object.MySQL >= 4.1 returns TIMESTAMP in the same format as DATETIME:>>> mysql_timestamp_converter('2007-02-25 22:32:17')datetime.datetime(2007, 2, 25, 22, 32, 17)MySQL < 4.1 uses a big string of numbers:>>> mysql_timestamp_converter('20070225223217')datetime.datetime(2007, 2, 25, 22, 32, 17)Illegal values are returned as None:>>> mysql_timestamp_converter('2007-02-31 22:32:17') is NoneTrue>>> mysql_timestamp_converter('00000000000000') is NoneTrue"""if not PY2 and isinstance(timestamp, (bytes, bytearray)):timestamp = timestamp.decode('ascii')if timestamp[4] == '-':return convert_datetime(timestamp)timestamp += "0"*(14-len(timestamp)) # paddingyear, month, day, hour, minute, second = \int(timestamp[:4]), int(timestamp[4:6]), int(timestamp[6:8]), \int(timestamp[8:10]), int(timestamp[10:12]), int(timestamp[12:14])try:return datetime.datetime(year, month, day, hour, minute, second)except ValueError:return Nonedef convert_set(s):if isinstance(s, (bytes, bytearray)):return set(s.split(b","))return set(s.split(","))def through(x):return x#def convert_bit(b):
#    b = "\x00" * (8 - len(b)) + b # pad w/ zeroes
#    return struct.unpack(">Q", b)[0]
#
#     the snippet above is right, but MySQLdb doesn't process bits,
#     so we shouldn't either
convert_bit = throughdef convert_characters(connection, field, data):field_charset = charset_by_id(field.charsetnr).nameencoding = charset_to_encoding(field_charset)if field.flags & FLAG.SET:return convert_set(data.decode(encoding))if field.flags & FLAG.BINARY:return dataif connection.use_unicode:data = data.decode(encoding)elif connection.charset != field_charset:data = data.decode(encoding)data = data.encode(connection.encoding)return dataencoders = {bool: escape_bool,int: escape_int,long_type: escape_int,float: escape_float,str: escape_str,text_type: escape_unicode,tuple: escape_sequence,list: escape_sequence,set: escape_sequence,frozenset: escape_sequence,dict: escape_dict,bytearray: escape_bytes,type(None): escape_None,datetime.date: escape_date,datetime.datetime: escape_datetime,datetime.timedelta: escape_timedelta,datetime.time: escape_time,time.struct_time: escape_struct_time,Decimal: escape_object,
}if not PY2 or JYTHON or IRONPYTHON:encoders[bytes] = escape_bytesdecoders = {FIELD_TYPE.BIT: convert_bit,FIELD_TYPE.TINY: int,FIELD_TYPE.SHORT: int,FIELD_TYPE.LONG: int,FIELD_TYPE.FLOAT: float,FIELD_TYPE.DOUBLE: float,FIELD_TYPE.LONGLONG: int,FIELD_TYPE.INT24: int,FIELD_TYPE.YEAR: int,FIELD_TYPE.TIMESTAMP: convert_mysql_timestamp,FIELD_TYPE.DATETIME: convert_datetime,FIELD_TYPE.TIME: convert_timedelta,FIELD_TYPE.DATE: convert_date,FIELD_TYPE.SET: convert_set,FIELD_TYPE.BLOB: through,FIELD_TYPE.TINY_BLOB: through,FIELD_TYPE.MEDIUM_BLOB: through,FIELD_TYPE.LONG_BLOB: through,FIELD_TYPE.STRING: through,FIELD_TYPE.VAR_STRING: through,FIELD_TYPE.VARCHAR: through,FIELD_TYPE.DECIMAL: Decimal,FIELD_TYPE.NEWDECIMAL: Decimal,
}

def _to_datetime(rs, col):java_val = rs.getTimestamp(col)if not java_val:returnd = datetime.datetime.strptime(str(java_val)[:19], "%Y-%m-%d %H:%M:%S")d = d.replace(microsecond=int(str(java_val.getNanos())[:6]))return str(d)def _to_time(rs, col):java_val = rs.getTime(col)if not java_val:returnreturn str(java_val)def _to_date(rs, col):java_val = rs.getDate(col)if not java_val:return# The following code requires Python 3.3+ on dates before year 1900.# d = datetime.datetime.strptime(str(java_val)[:10], "%Y-%m-%d")# return d.strftime("%Y-%m-%d")# Workaround / simpler soltution (see# https://github.com/baztian/jaydebeapi/issues/18):return str(java_val)[:10]def _to_binary(rs, col):java_val = rs.getObject(col)if java_val is None:returnreturn str(java_val)def _java_to_py(java_method):def to_py(rs, col):java_val = rs.getObject(col)if java_val is None:returnif PY2 and isinstance(java_val, (string_type, int, long, float, bool)):return java_valelif isinstance(java_val, (string_type, int, float, bool)):return java_valreturn getattr(java_val, java_method)()return to_py_to_double = _java_to_py('doubleValue')_to_int = _java_to_py('intValue')_to_boolean = _java_to_py('booleanValue')_DEFAULT_CONVERTERS = {# see# http://download.oracle.com/javase/8/docs/api/java/sql/Types.html# for possible keys'TIMESTAMP': _to_datetime,'TIME': _to_time,'DATE': _to_date,'BINARY': _to_binary,'DECIMAL': _to_double,'NUMERIC': _to_double,'DOUBLE': _to_double,'FLOAT': _to_double,'TINYINT': _to_int,'INTEGER': _to_int,'SMALLINT': _to_int,'BOOLEAN': _to_boolean,'BIT': _to_boolean
}

from .mysqldb import MySQLDialect_mysqldb
from ...util import langhelpers, py3kclass MySQLDialect_pymysql(MySQLDialect_mysqldb):driver = 'pymysql'description_encoding = None# generally, these two values should be both True# or both False.   PyMySQL unicode tests pass all the way back# to 0.4 either way.  See [ticket:3337]supports_unicode_statements = Truesupports_unicode_binds = Truedef __init__(self, server_side_cursors=False, **kwargs):super(MySQLDialect_pymysql, self).__init__(**kwargs)self.server_side_cursors = server_side_cursors@langhelpers.memoized_propertydef supports_server_side_cursors(self):try:cursors = __import__('pymysql.cursors').cursorsself._sscursor = cursors.SSCursorreturn Trueexcept (ImportError, AttributeError):return False@classmethoddef dbapi(cls):return __import__('pymysql')if py3k:def _extract_error_code(self, exception):if isinstance(exception.args[0], Exception):exception = exception.args[0]return exception.args[0]dialect = MySQLDialect_pymysql