a %0_&@sddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ddZ Gdd d e j Z Gd d d ejjjZdS) )absolute_importN)pycompat)indexapicCs6|dd}|dd}|dr2|ddd}|S)N_s\_%s\%*)replaceendswith)patternr @/usr/lib/python3/dist-packages/hgext/infinitepush/sqlindexapi.py_convertbookmarkpatterns    rcseZdZdZdfdd ZddZdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZZS) sqlindexapiz< Sql backend for infinitepush index. See schema.sql ,xc sttt|||_|||||d|_d|_d|_|sdS) NsADD BUNDLE %r %rs5INSERT INTO bundles(bundle, reponame) VALUES (%s, %s)paramsssINSERT INTO nodestobundle(node, bundle, reponame) VALUES (%s, %s, %s) ON DUPLICATE KEY UPDATE bundle=VALUES(bundle)s committerrscommitter_datesINSERT IGNORE INTO nodesmetadata(node, message, p1, p2, author, committer, author_date, committer_date, reponame) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s))rr0rinforrr/hexextrar"getintdateZ descriptionZp1Zp2) r!ZbundleidZnodesctxctxr>Z author_nameZcommitter_nameZ author_dateZcommitter_dater r r addbundle|s<  zsqlindexapi.addbundlecCsB|js||jd|j||f|jjd|||jfddS)zRTakes a bookmark name and hash, and records mapping in the metadata store.s&ADD BOOKMARKS %r bookmark: %r node: %rssINSERT INTO bookmarkstonode(bookmark, node, reponame) VALUES (%s, %s, %s) ON DUPLICATE KEY UPDATE node=VALUES(node)r:Nrr0rr<rrr/)r!bookmarknoder r r addbookmarks  zsqlindexapi.addbookmarkcCsh|js|g}g}t|D]$\}}|d||||jfq d|}|jj d||ddS)Ns (%s, %s, %s),siINSERT INTO bookmarkstonode(bookmark, node, reponame) VALUES %s ON DUPLICATE KEY UPDATE node=VALUES(node)r:) rr0rZ iteritemsappendextendrjoinrr/)r! bookmarksargsvaluesrErFr r raddmanybookmarkss  zsqlindexapi.addmanybookmarkscCsJ|js||jd||D]"}t|}|jjd||jfdq"dS)zAccepts 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. sDELETE BOOKMARKS: %ssFDELETE from bookmarkstonode WHERE bookmark LIKE (%s) and reponame = %sr:N)rr0rr<rrr/r)r!Zpatternsr r r rdeletebookmarksszsqlindexapi.deletebookmarkscCs|js||jd|j|f|jjd||jfd|j}t|dks`t|ddkrp|jddS|dd}|jd||S) zAReturns the bundleid for the bundle that contains the given node.sGET BUNDLE %r %rsBSELECT bundle from nodestobundle WHERE node = %s AND reponame = %sr:rrsNo matching nodeNsFound bundle %r rr0rr<rrr/fetchalllen)r!rFresultZbundler r r getbundles   zsqlindexapi.getbundlecCs|js||jd|j|f|jjd||jfd|j}t|dks`t|ddkrp|jddS|dd}|jd||S) zBReturns the node for the given bookmark. None if it doesn't exist.s"GET NODE reponame: %r bookmark: %rsFSELECT node from bookmarkstonode WHERE bookmark = %s AND reponame = %sr:rrsNo matching bookmarkNs Found node %rrQ)r!rErTrFr r rgetnodes     zsqlindexapi.getnodecCs|js||jd|j|ft|}|jjd|j|fd|j}i}|D]2}t |dkrv|jd|qT|d||d<qT|S)Ns&QUERY BOOKMARKS reponame: %r query: %rsSSELECT bookmark, node from bookmarkstonode WHERE reponame = %s AND bookmark LIKE %sr:sBad row returned: %srr) rr0rr<rrrr/rRrS)r!ZqueryrTrLrowr r r getbookmarkss$   zsqlindexapi.getbookmarkscCsB|js||jd|j||f|jjd||j|fddS)NsHINSERT METADATA, QUERY BOOKMARKS reponame: %r node: %r, jsonmetadata: %ssPUPDATE nodesmetadata SET optional_json_metadata=%s WHERE reponame=%s AND node=%sr:rD)r!rFZ jsonmetadatar r rsaveoptionaljsonmetadata s  z$sqlindexapi.saveoptionaljsonmetadata)rr)__name__ __module__ __qualname____doc__rr0r4r7r9rCrGrOrPrUrVrYrZ __classcell__r r r$rrs !$ 'rc@s.eZdZdZd ddZd ddZd ddZdS) r)ziEnsure that all values being returned are returned as python string (versus the default byte arrays).NcCst|Sr6strr!valueZdscr r r_STRING_to_python#sz!CustomConverter._STRING_to_pythoncCst|Sr6r`rbr r r_VAR_STRING_to_python&sz%CustomConverter._VAR_STRING_to_pythoncCst|Sr6r`rbr r r_BLOB_to_python)szCustomConverter._BLOB_to_python)N)N)N)r[r\r]r^rdrerfr r r rr)s  r))Z __future__rrrr,r1Zmysql.connectorr'Z mercurialrrrrr( conversionZMySQLConverterr)r r r rs