OwlCyberSecurity - MANAGER
Edit File: sql.pyc
� �t�ac @ s� d Z d d d g Z d d l Z y d d l Z Wn! e k rT d d l m Z n Xe j d � d d l m Z e � Z d Z e j Z d e f d � � YZ d S( s% Generic SQL database access object. t db_prefixt DatabaseErrort AbstractSQLDatabasei����N( t dbapi2s :memory:( t Loggers .db.c B s� e Z d Z e d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z e d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( s! A generic SQL database. c C s | | _ | | _ | | _ d S( s� @param dbpath: the path where the db file is stored. @type dbpath: str @param persistent: C{True} if the data in the DB must be perserved during upgrades, C{False} if the DB data can be re-created from an external source. @type persistent: bool @param autocommit: C{True} if auto-commit mode is desired, C{False} otherwise @type autocommit: bool N( t dbpatht persistentt autocommit( t selfR R R ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyt __init__3 s c C s d | j j | j f S( Ns <%s %r>( t __class__t __name__R ( R ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyt __repr__B s c C s t � d S( sE @return: the schema version assigned to this index. N( t NotImplementedError( R ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyt _db_versionE s c C s t � d S( sF @return: the collection type assigned to this index. N( R ( R ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyt _db_typeK s c C s� t | d � s�| j } y: | j r? t j | d d �| _ n t j | � | _ Wn' t k r{ t d | j f � � n X| j j � } | j d � z$| j | � r�| j | � \ } } | | j � k s� | | j � k r�| j � d } | j j � | ` | | j � k rTt j d d | d | d | j � �t j | � | j � S| | j � k r�t j d d | d | d | j � �| j | � Sq�n | j | | � | j j � Wd | d k r�| j � n Xn | j S( s� Access the underlying database. @return: a db2 connection object for this index's underlying data store. t _db_connectiont isolation_levels Unable to open database %ss PRAGMA journal_mode = PERSISTs/ Database {f} has different type ({t1} vs. {t2})t ft t1t t2s5 Database {f} has different schema (v.{v1} vs. v.{v2})t v1t v2N( t hasattrR R t sqlitet connectt NoneR R t cursort executet _test_schema_tablet _get_schema_versionR R t closet logt errort ost removet _dbt _db_upgradet _db_initt commit( R t db_filenamet qt versiont dbtype( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR$ Q sD $ c C s | j d � | j � S( Nsa select (1) from SQLITE_MASTER where TYPE = 'table' and NAME = 'CALDAV' ( R t fetchone( R R) ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR � s c C sn | j d � | j � } | d k r2 | d } n | j d � | j � } | d k rd | d } n | | f S( Ns\ select VALUE from CALDAV where KEY = 'SCHEMA_VERSION' i sR select VALUE from CALDAV where KEY = 'TYPE' ( R R, R ( R R) R* R+ ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR � s c C s� t j d d | �| j j } d | j _ yW | j d � | j | � st | j | � | j | � | j t � n | j d � Wn t k r� n X| | j _ d S( s� Initialise the underlying database tables. @param db_filename: the file name of the index database. @param q: a database cursor to use. s Initializing database {f}R s begin exclusive transactionR' N( R t infoR R R R R t _db_init_schema_tablet _db_init_data_tablest _db_recreatet FalseR ( R R( R) t old_isolation( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR& � s c C sC | j d � | j d | j � g � | j d | j � g � d S( s� Initialise the underlying database tables. @param db_filename: the file name of the index database. @param q: a database cursor to use. sp create table CALDAV ( KEY text unique, VALUE text unique ) sc insert into CALDAV (KEY, VALUE) values ('SCHEMA_VERSION', :1) sY insert into CALDAV (KEY, VALUE) values ('TYPE', :1) N( R R R ( R R) ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR. � s c C s t � d S( s� Initialise the underlying database tables. @param db_filename: the file name of the index database. @param q: a database cursor to use. N( R ( R R) ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR/ � s c C s | r | j � n d S( s/ Recreate the database tables. N( t _db_commit( R t do_commit( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR0 � s c C s | j S( N( R ( R t old_version( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyt _db_can_upgrade s c C s� | j | � r� t j | j d d �| _ | j j � } | j d � | j | � \ } } | | j � k r� | j | | � | j | � n | j d � | j � n t j | j � | j � S( s. Upgrade the database tables. R s begin exclusive transactionR' N( R6 R R R R R R R R R t _db_upgrade_data_tablest _db_upgrade_schemat _db_closeR" R# R$ ( R R5 R) R* t _ignore_dbtype( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR% s c C s t d � � d S( sC Upgrade the data from an older version of the DB. s4 Persistent databases MUST support an upgrade method.N( R ( R R) R5 ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR7 s c C s | j d | j � g � d S( sG Upgrade the stored schema version to the current one. sn insert or replace into CALDAV (KEY, VALUE) values ('SCHEMA_VERSION', :1) N( R R ( R R) ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR8 ( s c C s) t | d � r% | j j � | ` n d S( NR ( R R R ( R ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR9 3 s c G s d � | j | | � D� S( sw Execute an SQL query and obtain the resulting values. @param sql: the SQL query to execute. @param query_params: parameters to C{sql}. @return: an interable of values in the first column of each row resulting from executing C{sql} with C{query_params}. @raise AssertionError: if the query yields multiple columns. c s s | ] } | d Vq d S( i N( ( t .0t row( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pys <genexpr>A s ( t _db_execute( R t sqlt query_params( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyt _db_values_for_sql8 s c G sO d } xB | j | | � D]. } | d k sA t d | | f � � | } q W| S( sP Execute an SQL query and obtain a single value. @param sql: the SQL query to execute. @param query_params: parameters to C{sql}. @return: the value resulting from the executing C{sql} with C{query_params}. @raise AssertionError: if the query yields multiple rows or columns. s Multiple values in DB for %s %sN( R R@ t AssertionError( R R> R? t valueR< ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyt _db_value_for_sqlC s " c G s� | j � j � } zd y* | j | | � | j | _ | j � SWn3 t k rt t j d d | d | d | �� n XWd | j � Xd S( s Execute an SQL query and obtain the resulting values. @param sql: the SQL query to execute. @param query_params: parameters to C{sql}. @return: an interable of tuples for each row resulting from executing C{sql} with C{query_params}. s6 Exception while executing SQL on DB {s!r}: {q!r} {p!r}t sR) t pN( R$ R R t lastrowidt fetchallR R R! R ( R R> R? R) ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR= R s c C s | j j � d S( N( R R' ( R ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR3 e s c C s | j j � d S( N( R t rollback( R ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyt _db_rollbackh s ( R t __module__t __doc__R1 R R R R R$ R R R& R. R/ t TrueR0 R6 R% R7 R8 R9 R@ RC R= R3 RI ( ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyR . s, E ( RK t __all__R" t sqlite3R t ImportErrort pysqlite2R R t twext.python.logR R R R t objectR ( ( ( s$ /opt/cpanel-ccs/twistedcaldav/sql.pyt <module> s