mirror of
https://github.com/facebook/sapling.git
synced 2024-10-12 01:39:21 +03:00
e3c0c7ab65
Summary: Common library to interact with mysql. Will be used in the next diffs to record pushrebase requests Reviewed By: quark-zju Differential Revision: D8801603 fbshipit-source-id: 1a926742cc40458e8f9da385620f0f3837f3ae09
53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
# mysqlutil.py - useful utility methods for accessing mysql from server-side hg
|
|
#
|
|
# Copyright 2017 Facebook, Inc.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2 or any later version.
|
|
|
|
|
|
class InvalidConnectionString(Exception):
|
|
pass
|
|
|
|
|
|
def parseconnectionstring(connstr):
|
|
"""
|
|
Parses connection string in format 'IP:PORT:DB_NAME:USER:PASSWORD' and return
|
|
parameters for mysql.connection module
|
|
"""
|
|
|
|
try:
|
|
host, port, db, user, password = connstr.rsplit(":", 4)
|
|
return {
|
|
"host": host,
|
|
"port": port,
|
|
"database": db,
|
|
"user": user,
|
|
"password": password,
|
|
}
|
|
except ValueError:
|
|
raise InvalidConnectionString()
|
|
|
|
|
|
def insert(sqlconn, tablename, argsdict):
|
|
"""
|
|
Inserts new row into a table, given a name of a table and a mapping
|
|
column name -> column value
|
|
"""
|
|
|
|
sqlcursor = sqlconn.cursor()
|
|
|
|
items = list(argsdict.items())
|
|
columns = ", ".join(
|
|
("{column_name}".format(column_name=column_name) for column_name, _ in items)
|
|
)
|
|
|
|
placeholders = ", ".join(("%s" for _ in items))
|
|
|
|
insertstmt = "INSERT INTO {table} ({columns}) VALUES ({placeholders})".format(
|
|
table=tablename, columns=columns, placeholders=placeholders
|
|
)
|
|
|
|
sqlcursor.execute(insertstmt, params=[value for _, value in items])
|
|
sqlconn.commit()
|