�
��Vc@spdd
lZddl
mZ
ddlmZ
ydd
lZWn


n
XddlmZ
def
d��YZ	d
S(i����N(
texc_info(
tLocki
(
t
BaseFoldertLocalStatusSQLiteFoldercBs
eZ
dZd
Zd�Zd�Zd�Zd�Zd�Zde
d�Zd�Zd	�Z
d
�Zd�Zd�Zd
�Zd�Zd�Zde�d�Zd�Zd�Zdd�
Zd�Zd�Zd�Zd�Zd�Zd�Zd�Zd�Z RS(sn
LocalStatus backend implemented with an SQLite database

    As python-sqlite currently does not allow to access the same sqlite
    objects from various threads, we need to open get and close a db
    connection and cursor for all operations. This is a big disadvantage
    and we might want to investigate if we cannot hold an object open
    for a thread somehow.icCs�
d
|_t
t|�j|
|�
|j|_tjj|j�|j	��|_
i|_t|_
tjj|j
�
}tjj|�
s�tj|�

ntjj|�
s�td|�
�
nt�|_ytj|j
dt�

|_Wnltk
r



td|j
�
dt�d�n<tjk
rY

}
td|j
|f�
dt�d�n
Xtjdksu
td�
�
y|jjd	�
}Wntjk
r�



|j �
n6Xt!|j"�d
�
}|tj#kr�
|j$|�

ndS(Nt
.s-SQLite database path '%s' is not a directory.tcheck_same_threadsmSQLite backend chosen, but cannot connect with available bindings to '%s'. Is the sqlite3 package installed?.is�cannot open database file '%s': %s.
You might want to check the rights to that file and if it cleanly opens with the 'sqlite<3>' command.i
s'Your sqlite is not multithreading safe.s1SELECT value from metadata WHERE key='db_version'i(%tseptsuperRt__init__troottostpathtjointgetroottgetfolderbasenametfilenametmessagelisttFalset
_newfoldertdirnametexiststmakedirstisdirtUserWarningR
t_dblocktsqlitetconnectt
connectiont	NameErrortNoneRtOperationalErrortthreadsafetytAssertionErrortexecutet
DatabaseErrort#_LocalStatusSQLiteFolder__create_dbtinttfetchonetcur_versiont$_LocalStatusSQLiteFolder__upgrade_db(tselftnamet
repositoryRt
etcursortversion((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pyR.s<
	


$
		









$
	




c


CstS(
N(
R(
R(((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytstoresmessagesas
c


Cs|jS(
N(
R(
R(((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytgetfullnameds
c


Cs|jS(
N(
R(
R(((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytisnewfolderhs
c

Cs|jd
�

dS(sDelete all messages in the db.sDELETE FROM statusN(
t#_LocalStatusSQLiteFolder__sql_write(
R(((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytdeletemessagelistmscCs 
t}x
|s
|j
j�
z�y�|dkr^|rI|jj|
�
}q�|jj|
�
}n3|r||jj|
|�}n|jj|
|�}t}|jj�
Wn^t	j
k
r

}
|jd
dkr�q
|jd
dkr�|jj
dd�
t}q
�n
XWd|j
j�
Xq	W|S(s�
Execute some SQL, retrying if the db was locked.

        :param sql: the SQL string passed to execute()
        :param vars: the variable values to `sql`. E.g. (1,2) or {uid:1,
            flags:'T'}. See sqlite docs for possibilities.
        :param executemany: bool indicating whether we want to
            perform conn.executemany() or conn.execute().
        :returns: the Cursor() or raises an Exceptionis(cannot commit - no transaction is activesdatabase is lockedts!Locked sqlite database, retrying.N(RRtacquireRRtexecutemanyR!tTruetcommitRRtargstuitdebugtrelease(R(tsqltvarsR5tsuccessR,R+((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pyt__sql_writess.

	














	
cCs�t|d
�r|j
j�
ntj|jdt�

|_
|
dk
r�|jjd|j	|f�

|j
j
d�

|j
j�
ndS(s<Upgrade the sqlite format from version 'from_ver' to currentRRi
sAUpgrading LocalStatus cache from version 1 to version 2 for %s:%ss!
ALTER TABLE status ADD mtime INTEGER DEFAULT 0;
                                             ALTER TABLE status ADD labels VARCHAR(256) DEFAULT '';
                                             UPDATE metadata SET value='2' WHERE key='db_version';
                                          N(thasattrRtcloseRRRRR9t_msgR*t
executescriptR7(R(tfrom_ver((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pyt__upgrade_db�s





	
c

CsG|jj
d
|j|f�

|jjd�

|jj�
t|_dS(suCreate a new db file.

        self.connection must point to the opened and valid SQlite
        database connection.s&Creating new Local Status db for %s:%ss�
        CREATE TABLE metadata (key VARCHAR(50) PRIMARY KEY, value VARCHAR(128));
        INSERT INTO metadata VALUES('db_version', '2');
        CREATE TABLE status (id INTEGER PRIMARY KEY, flags VARCHAR(50), mtime INTEGER, labels VARCHAR(256));
        N(R9RBR*RRCR7R6R(
R(((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pyt__create_db�s

	


cCs-i|
d
6t�d6t�d6dd6dd6S(Ntuidtflagstlabelsittimetmtime(
tset(R(RG((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytmsglist_item_initializer�s
c
Cs
i|_|j
jd
�
}
x�|
D]�}|d}|j|�
|j|<t|d�
}yNtg|djd�
D]*}t|j��
dkrr|j�^qr�
}Wntk
r�


t�}n
X||j|d<||j|d<|d|j|d	<q"WdS(
Ns(SELECT id,flags,mtime,labels from statusii
it
,RHRIiRK(	RRR!RMRLtsplittlentstriptAttributeError(R(R,trowRGRHtlbRI((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytcachemessagelist�s
	









H





c

Cs
i|_dS(
N(
R(
R(((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytdropmessagelistcache�s
c


CsdS(
N((
R(((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytsave�s
c
Cs�g}
xr|jj
�D]a\}}|d
}djt|d�
�
}djt|d�
�
}|
j||||f�

qW|jd|
dt�

dS(	s-Saves the entire messagelist to the database.RKR3RHs, RIsFINSERT OR REPLACE INTO status (id,flags,mtime,labels) VALUES (?,?,?,?)R5N(RtitemsRtsortedtappendR1R6(R(tdataRGtmsgRKRHRI((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytsaveall�s





	c


Cs|jS(
N(
R(
R(((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytgetmessagelist
s
icCs�|
d
kr|
S|j|
�
r3|j
|
|�
|
S|j|
�
|j|
<i|
d6|d6|d6|d6|d6|j|
<djt|�
�
}djt|�
�
}|jd	|
|||f�
|
S(
s�Writes a new message, with the specified uid.

        See folder/Base for detail. Note that savemessage() does not
        check against dryrun settings, so you need to ensure that
        savemessage is never called in a dryrun mode.iRGRHRJRKRIR3s, s;INSERT INTO status (id,flags,mtime,labels) VALUES (?,?,?,?)(t	uidexiststsavemessageflagsRMRRRYR1(R(RGtcontentRHtrtimeRKRI((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytsavemessage#
s


0


	

cCsU|j|
�
st
�
||j|
d
<djt|�
�
}|jd||
f�
dS(NRHR3s$UPDATE status SET flags=? WHERE id=?(R_R RRRYR1(R(RGRH((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pyR`<
s



cCs|j|
d
S(NRH(
R(R(RG((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytgetmessageflagsC
s
cCs|||j|
d
<|r+||j|
d<ndj
t|�
�
}|rb|jd|||
f�
n|jd||
f�
dS(NRIRKs, s.UPDATE status SET labels=?, mtime=? WHERE id=?s%UPDATE status SET labels=? WHERE id=?(RRRYR1(R(RGRIRK((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytsavemessagelabelsG
s



cCs�g|
j�D]'\}}d
j
t|�
�
|f^q
}|jd|dt�

x+|
j�D]\}}||j|d<q]WdS(sY
        Saves labels from a dictionary in a single database operation.
        
        s, s%UPDATE status SET labels=? WHERE id=?R5RIN(RXRRYR1R6R(R(RIRGt
lR[((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytsavemessageslabelsbulkR
s:


cCs�g}xE|
D]=}|j|d
|B}|j
djt|�
�
|f�

q
W|jd|dt�

x.|
D]&}|j|d
|B|j|d
<qkWdS(NRIs, s%UPDATE status SET labels=? WHERE id=?R5(RRZRRYR1R6(R(tuidsRIR[RGt	newlabels((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytaddmessageslabels]
s




&



cCs�g}xE|
D]=}|j|d
|}|j
djt|�
�
|f�

q
W|jd|dt�

x.|
D]&}|j|d
||j|d
<qkWdS(NRIs, s%UPDATE status SET labels=? WHERE id=?R5(RRZRRYR1R6(R(RhRIR[RGRi((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytdeletemessageslabelsg
s




&



cCs|j|
d
S(NRI(
R(R(RG((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytgetmessagelabelsq
s
cCssg|
j�D]\}}||f^q
}|j
d
|dt�

x+|
j�D]\}}||j|d<qNWdS(sGSaves mtimes from the mtimes dictionary in a single database operation.s$UPDATE status SET mtime=? WHERE id=?R5RKN(RXR1R6R(R(tmtimesRGtmtR[((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytsavemessagesmtimebulku
s+


cCs|j|
d
S(NRK(
R(R(RG((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytgetmessagemtime~
s
cCs4|
|jkrdS|j
d
|
f
�
|j|
=dS(NsDELETE FROM status WHERE id=?(RR1(R(RG((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pyt
deletemessage�
s



cCspg|
D]}||jkr|^q}
t
|
�
s8d
S|jdt|
�
t�
x|
D]}|j|=qXWd
S(s�Delete list of UIDs from status cache

        This function uses sqlites executemany() function which is
        much faster than iterating through deletemessage() when we have
        many messages to delete.NsDELETE FROM status WHERE id=?(RRPR1tzipR6(R(tuidlistRG((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pytdeletemessages�
s(




N(!t__name__t
__module__t__doc__R&RR.R/R0R2RRR1R'R#RMRURVRWR]R^RLRcR`RdReRgRjRkRlRoRpRqRt(((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pyRs8		3				&							5					
	
					(
R
tsysRt	threadingR
tsqlite3RtBaseRR(((sH/usr/lib/python2.7/site-packages/offlineimap/folder/LocalStatusSQLite.pyt<module>s