B dv'@sdZddlZddlZddlmZdZejZejZ Gdddej Z e de Gddde ZGd d d eZGd d d ZGd ddZdS)a|lightweight database dumper Copyright (C) 2003 Duncan Brown This file is part of the lightweight datapase dumper (ldbd) The ldbd module provides classes for manipulating LIGO metadata database tables. References: http://www.ligo.caltech.edu/docs/T/T990101-02.pdf http://www.ligo.caltech.edu/docs/T/T990023-01.pdf http://ldas-sw.ligo.caltech.edu/doc/db2/doc/html/ilwdformat.html N) git_versionz&Duncan Brown c@s.eZdZdZdZdZdZdZdZe j Z dZ dS) LIGOLWStreamz= Create a csv parser dialect for parsing LIGO_LW streams ,F\ "TN) __name__ __module__ __qualname____doc__ delimiter doublequote escapecharlineterminator quotecharcsv QUOTE_ALLquotingskipinitialspacerrV/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/glue/ldbd.pyr)src@seZdZdZdS)LIGOLwParseErrorz'Error parsing LIGO lightweight XML fileN)rr r r rrrrr8src@s(eZdZdZddZddZddZdS) XlatorzL All in one multiple string substitution class from the python cookbook cCstdttj|S)zC Build a re object based on keys in the current dictionary |)recompilejoinmapescapekeys)selfrrr _make_regexAszXlator._make_regexcCs||dS)z. Handler invoked for each regex match r)group)r matchrrr__call__GszXlator.__call__cCs|||S)z3 Translate text, returns the modified text )r!sub)r textrrrxlatMsz Xlator.xlatN)rr r r r!r$r'rrrrr=src@s0eZdZdZddZddZddZdd Zd S) LIGOLwParserzr Provides methods for parsing the data from a LIGO lightweight XML file parsed with pyRXP into a dictionary cCstd|_td|_td|_td|_td|_td|_td|_td|_ d |_ t t t t t t t t t t |j |j|jd |_td d d dd|_d S)z Initializes a LIGO lightweight XML parser with the necessary regular expressions and function for tuple translation z&(\A[a-z0-9_]+:|\A)([a-z0-9_]+):table\Zz0(\A[a-z0-9_]+:|\A)([a-z0-9_]+:|\A)([a-z0-9_]+)\Zz\A\s*"z"\s*\Zz\A\s+"z"*\s*\Zz\A\\[0-9][0-9][0-9]z\\,N) Zint2sZint_2sZint4sZint_4sZint8sZint_8sZreal4Zreal_4Zreal8Zreal_8Zlstringz ilwd:charz ilwd:char_u&>td|dn:y|j|}Wn(tk rv|jsj|}nt dYnX|S)z If the ilwd:char field contains octal data, it is translated to a binary string and returned. Otherwise a lookup is done in the unique id dictionary and a binary string containing the correct unique id is returned. r<zistr = ''z(unique id table has not been initialized) r0r%r1r2r#execr4lookupAttributeErrorr)r ZistrZ istr_origrrrZ __ilwdchars zLIGOLwParser.__ilwdcharcsi}d}x^|dD]P}|ddks4|ddkr^|dd}y|j|d}Wn tk rztd|YnX|idd d ||<x|dD]}|dd ks|dd kr|dd}y|j|d }Wn tk rtd|YnXy|dd}Wn"tk r2td|YnX|||d |<||dg |qW|d7}qWxF| D]8}x.|d||ddD]}|ddks|ddkry|dd} Wntk rtdYnX| dkrtd| t |ddkr |d d |dd} | | | rN| dt |  } t ttdd | gt} t | } t ||d t| \} }|dkrtdfddt| D}xt| D]}t|\}}yZ||d|}t | |dkrd|||<n&|j||d || ||||<Wn\ttfk r}z8dt|}|d||| |t||f7}t|Wdd}~XYnXqWttt|||d<qWqvW|S) z Parse an XML tuple returned by pyRXP into a dictionary of LIGO metadata elements. The dictionary contains one entry for each table found in the XML tuple. rTabletableNamez#unable to parse a valid table name rr<)poscolumnstreamqueryColumnrGz$unable to parse a valid column name Typeztype is missing for column Z orderedcolrFZStreamrH Delimiterzstream is missing delimiterrz#unable to handle stream delimiter: Nz\nz'invalid stream length for given columnscsg|]}dgqS)Nr).0i)ntyprr sz+LIGOLwParser.parsetuple..zstream translation error (%s) z#for column %s in table %s: %s -> %s)lowerr,r#r"r@rr-KeyError setdefaultappendrlenendswithnextrreaderrr%rdivmodranger9 ValueErrorstrlistrtuple)r ZxmltuplerCZtupleidxtagtabZsubtagcoltypdelimrawrHslenmlenZlftlstrOjkthiscolerrmsgmsgr)rPr parsetuples|  "    *  "zLIGOLwParser.parsetupleN)rr r r r;r7r8rnrrrrr(Ts  r(c@s"eZdZdZdddZddZdS) LIGOMetadataz LIGO Metadata object class. Contains methods for parsing a LIGO lightweight XML file and inserting it into a database, executing and SQL query to retrive data from the database and writing it to a LIGO lightweight XML file NcCs8||_||_|rd|j_i|_tdddddd|_dS)z Connects to the database and creates a cursor. Initializes the unique id table for this LIGO lw document. xmlparser = pyRXP XML to tuple parser object lwtparser = LIGOLwParser object (tuple parser) Nz&z>z<z\\z\")r)r*r+rr) xmlparser lwtparserr4rCrZstrtoxml)r rprqrrrr;s zLIGOMetadata.__init__cCsX|jstd|jstdddd|dD}||d}|j||_dS) zz Parses an XML document into a form read for insertion into the database xml = the xml document to be parsed zpyRXP parser not initializedz$LIGO_LW tuple parser not initializedr<cSsg|] }|qSr)strip)rNxrrrrQsz&LIGOMetadata.parse..rzutf-8N)rprrqrsplitencodernrC)r xmlZ ligolwtuprrrparseszLIGOMetadata.parse)NN)rr r r r;rwrrrrros ro)r rrZgluer __author__date__date__id __version__Dialectrregister_dialect Exceptionrdictrr(rorrrrs  ,