|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/hgext/infinitepush/__pycache__/ |
Upload File : |
o
�]Lb�& � @ 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�r| d d� d } | S )N� _s \_� %s \%� *���)�replace�endswith)�pattern� r
�@/usr/lib/python3/dist-packages/hgext/infinitepush/sqlindexapi.py�_convertbookmarkpattern s
r c s� e Zd ZdZ d � 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�logging� basicConfig� getLogger�log�setLevel�
_connected� _waittimeout� _locktimeout) �selfr �host�port�database�user�password�logfile�loglevel� waittimeout� locktimeout�� __class__r
r r # s$
�
zsqlindexapi.__init__c C s� | j rt�d��| jrt�d��d} ztjjd i | j��| _ | j �t � d| j _
W n tjj j yD |d8 }|dkr=� t
�d� Y nw 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� TFr r g�������?s %ss SET wait_timeout=%ss SET innodb_lock_wait_timeout=%sr
)r r �indexexceptionr �mysql� connector�connectr �set_converter_class�CustomConverter�
autocommit�errors�Error�time�sleep� converter�escaper" �cursor�executer# r! )r$ �retryr, r
r
r �
sqlconnectD s6
� �� �
zsqlindexapi.sqlconnectc C sV t �� � t �d� | j�� | j�� W d � n1 s w Y d| _d| _dS )z(Cleans up the metadata store connection.s ignoreN)�warnings�catch_warnings� simplefilterr �closer �r$ r
r
r rE h s
�
zsqlindexapi.closec C s | j s| �� | S �N)r! rA rF r
r
r � __enter__q s zsqlindexapi.__enter__c C s$ |d u r | j �� d S | j �� d S rG )r �commit�rollback)r$ �exc_type�exc_val�exc_tbr
r
r �__exit__v s zsqlindexapi.__exit__c C s� | j s| �� | j�d| j|f � | jjd|| jfd� |D ]N}| 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! rA r �infor r r? �hex�extrar( �get�int�date� description�p1�p2) r$ �bundleid�nodesctx�ctxrS � author_name�committer_name� author_date�committer_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)rO N�r! rA r rQ r r r? )r$ �bookmark�noder
r
r � addbookmark� s
��
�zsqlindexapi.addbookmarkc C sh | j s| �� g }g }t�|�D ]\}}|�d� |�||| jf� qd�|�}| 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)rO )
r! rA r � iteritems�append�extendr �joinr r? )r$ � bookmarks�args�valuesrc rd r
r
r �addmanybookmarks� s
�
�z sqlindexapi.addmanybookmarksc C sJ | j s| �� | j�d| � |D ]}t|�}| jjd|| jfd� qdS )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 = %srO N)r! rA r rQ r r r? r )r$ �patternsr r
r
r �deletebookmarks� s ��zsqlindexapi.deletebookmarksc C � | j s| �� | j�d| j|f � | jjd|| jfd� | j�� }t|�dks0t|d �dkr8| 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 = %srO r r s No matching nodeNs Found bundle %r� r! rA r rQ r r r? �fetchall�len)r$ rd �result�bundler
r
r � getbundle� s �
zsqlindexapi.getbundlec C rq ) 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 = %srO r r s No matching bookmarkNs
Found node %rrr )r$ rc ru rd r
r
r �getnode� s ��
zsqlindexapi.getnodec C s� | j s| �� | j�d| j|f � t|�}| jjd| j|fd� | j�� }i }|D ]}t |�dkr;| j�d| � q*|d ||d < q*|S )Ns& QUERY BOOKMARKS reponame: %r query: %rsS SELECT bookmark, node from bookmarkstonode WHERE reponame = %s AND bookmark LIKE %srO � s Bad row returned: %sr r )
r! rA r rQ r r r r? rs rt )r$ �queryru rk �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=%srO rb )r$ rd � jsonmetadatar
r
r �saveoptionaljsonmetadata
s
��
�z$sqlindexapi.saveoptionaljsonmetadata)r r )�__name__�
__module__� __qualname__�__doc__r rA rE rH rN ra re rn rp rw rx r| r~ �
__classcell__r
r
r. r r s" �!$ 'r c @ s. e Zd ZdZd dd�Zd dd�Zd dd�ZdS )
r6 ziEnsure that all values being returned are returned as python string
(versus the default byte arrays).Nc C � t |�S rG ��str�r$ �value�dscr
r
r �_STRING_to_python# � z!CustomConverter._STRING_to_pythonc C r� rG r� r� r
r
r �_VAR_STRING_to_python& r� z%CustomConverter._VAR_STRING_to_pythonc C r� rG r� r� r
r
r �_BLOB_to_python) r� z CustomConverter._BLOB_to_pythonrG )r r� r� r� r� r� r� r
r
r
r r6 s
r6 )�
__future__r r r r: rB �mysql.connectorr2 � mercurialr � r r r r3 �
conversion�MySQLConverterr6 r
r
r
r �<module> s