ruạṛ
a �+�b�& � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlm Z ddl m Z dd� Z G dd � d e j �Z G d d � d ejjj�ZdS ) � )�absolute_importN)�pycompat� )�indexapic C s6 | � dd�} | � dd�} | �d�r2| d d� d } | S )N� _s \_� %s \%� *���)�replace�endswith)�pattern� r �D/usr/lib64/python3.9/site-packages/hgext/infinitepush/sqlindexapi.py�_convertbookmarkpattern s r c s� e Zd ZdZd � fdd� Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd d � Z� ZS ) � sqlindexapiz< Sql backend for infinitepush index. See schema.sql �, �x c st t t| ��� || _|||||d�| _d | _d | _|s<tj}t j |d� t � � | _ | j � |� d| _| | _| | _d S )N)s hosts ports databases users password)�filenameF)�superr �__init__�reponame�sqlargs�sqlconn� sqlcursor�os�devnull�loggingZ basicConfigZ getLogger�logZsetLevel� _connected� _waittimeout� _locktimeout) �selfr �host�portZdatabase�user�passwordZlogfileZloglevel� waittimeoutZ locktimeout�� __class__r r r # s$ � zsqlindexapi.__init__c C s� | j rt�d��| jr t�d��d}z2tjjf i | j��| _ | j �t � d| j _ W q�W q$ tjj j y� |d8 }|dkr|� t �d� Y q$0 q$| j j�d| j �}| j �� | _| j�d | � | j�d | j � d | _d S ) Ns SQL connection already opens* SQL cursor already open without connection� Fr r g�������?s %ss SET wait_timeout=%ss SET innodb_lock_wait_timeout=%sT)r r Zindexexceptionr �mysql� connectorZconnectr Zset_converter_class�CustomConverterZ autocommit�errors�Error�time�sleepZ converter�escaper Zcursor�executer r )r! Zretryr& r r r � sqlconnectD s0 � �zsqlindexapi.sqlconnectc C sV t �� �. t �d� | j�� | j�� W d � n1 s<0 Y d| _d| _dS )z(Cleans up the metadata store connection.s ignoreN)�warnings�catch_warnings� simplefilterr �closer �r! r r r r7 h s (zsqlindexapi.closec C s | j s| �� | S �N)r r3 r8 r r r � __enter__q s zsqlindexapi.__enter__c C s" |d u r| j �� n | j �� d S r9 )r ZcommitZrollback)r! �exc_typeZexc_valZexc_tbr r r �__exit__v s zsqlindexapi.__exit__c C s� | j s| �� | j�d| j|f � | jjd|| jfd� |D ]�}| jjd|�� || jfd� |�� }|� � }|� d|� � �}t |� � d �}t |� d|��}| jjd|�� |� � |�� �� |�� �� ||||| jf d� q>d S ) Ns ADD BUNDLE %r %rs5 INSERT INTO bundles(bundle, reponame) VALUES (%s, %s)��paramsss INSERT INTO nodestobundle(node, bundle, reponame) VALUES (%s, %s, %s) ON DUPLICATE KEY UPDATE bundle=VALUES(bundle)s committerr s committer_dates� INSERT IGNORE INTO nodesmetadata(node, message, p1, p2, author, committer, author_date, committer_date, reponame) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s))r r3 r �infor r r2 �hex�extrar$ �get�int�date� descriptionZp1Zp2) r! ZbundleidZnodesctx�ctxrA Z author_nameZcommitter_nameZ author_dateZcommitter_dater r r � addbundle| s< �� ��zsqlindexapi.addbundlec C sB | j s| �� | j�d| j||f � | jjd||| jfd� dS )zRTakes a bookmark name and hash, and records mapping in the metadata store.s& ADD BOOKMARKS %r bookmark: %r node: %rss INSERT INTO bookmarkstonode(bookmark, node, reponame) VALUES (%s, %s, %s) ON DUPLICATE KEY UPDATE node=VALUES(node)r= N�r r3 r r? r r r2 )r! �bookmark�noder r r � addbookmark� s �� �zsqlindexapi.addbookmarkc C sh | j s| �� g }g }t�|�D ]$\}}|�d� |�||| jf� q d�|�}| jj d| |d� d S )Ns (%s, %s, %s)� ,si INSERT INTO bookmarkstonode(bookmark, node, reponame) VALUES %s ON DUPLICATE KEY UPDATE node=VALUES(node)r= ) r r3 r Z iteritems�append�extendr �joinr r2 )r! � bookmarks�args�valuesrI rJ r r r �addmanybookmarks� s ��z sqlindexapi.addmanybookmarksc C sJ | j s| �� | j�d| � |D ]"}t|�}| jjd|| jfd� q"dS )z�Accepts list of bookmark patterns and deletes them. If `commit` is set then bookmark will actually be deleted. Otherwise deletion will be delayed until the end of transaction. s DELETE BOOKMARKS: %ssF DELETE from bookmarkstonode WHERE bookmark LIKE (%s) and reponame = %sr= N)r r3 r r? r r r2 r )r! �patternsr r r r �deletebookmarks� s �zsqlindexapi.deletebookmarksc C s� | j s| �� | j�d| j|f � | jjd|| jfd� | j�� }t|�dks`t|d �dkrp| j�d� dS |d d }| j�d| � |S ) zAReturns the bundleid for the bundle that contains the given node.s GET BUNDLE %r %rsB SELECT bundle from nodestobundle WHERE node = %s AND reponame = %sr= r r s No matching nodeNs Found bundle %r� r r3 r r? r r r2 �fetchall�len)r! rJ �resultZbundler r r � getbundle� s � zsqlindexapi.getbundlec C s� | j s| �� | j�d| j|f � | jjd|| jfd� | j�� }t|�dks`t|d �dkrp| j�d� dS |d d }| j�d| � |S ) zBReturns the node for the given bookmark. None if it doesn't exist.s" GET NODE reponame: %r bookmark: %rsF SELECT node from bookmarkstonode WHERE bookmark = %s AND reponame = %sr= r r s No matching bookmarkNs Found node %rrV )r! rI rY rJ r r r �getnode� s �� zsqlindexapi.getnodec C s� | j s| �� | j�d| j|f � t|�}| jjd| j|fd� | j�� }i }|D ]2}t |�dkrv| j�d| � qT|d ||d < qT|S )Ns& QUERY BOOKMARKS reponame: %r query: %rsS SELECT bookmark, node from bookmarkstonode WHERE reponame = %s AND bookmark LIKE %sr= � s Bad row returned: %sr r ) r r3 r r? r r r r2 rW rX )r! �queryrY rP �rowr r r � getbookmarks� s$ �� zsqlindexapi.getbookmarksc C sB | j s| �� | j�d| j||f � | jjd|| j|fd� d S )NsH INSERT METADATA, QUERY BOOKMARKS reponame: %r node: %r, jsonmetadata: %ssP UPDATE nodesmetadata SET optional_json_metadata=%s WHERE reponame=%s AND node=%sr= rH )r! rJ Z jsonmetadatar r r �saveoptionaljsonmetadata s �� �z$sqlindexapi.saveoptionaljsonmetadata)r r )�__name__� __module__� __qualname__�__doc__r r3 r7 r: r< rG rK rS rU rZ r[ r_ r` � __classcell__r r r' r r s �!$ 'r c @ s. e Zd ZdZd dd�Zd dd�Zd dd�ZdS ) r, ziEnsure that all values being returned are returned as python string (versus the default byte arrays).Nc C s t |�S r9 ��str�r! �valueZdscr r r �_STRING_to_python# s z!CustomConverter._STRING_to_pythonc C s t |�S r9 rf rh r r r �_VAR_STRING_to_python& s z%CustomConverter._VAR_STRING_to_pythonc C s t |�S r9 rf rh r r r �_BLOB_to_python) s z CustomConverter._BLOB_to_python)N)N)N)ra rb rc rd rj rk rl r r r r r, s r, )Z __future__r r r r/ r4 Zmysql.connectorr* Z mercurialr � r r r r+ � conversionZMySQLConverterr, r r r r �<module> s
cải xoăn