ó BÜXQc@s|dZddlZddlZddlmZmZmZddlmZm Z ej dƒj Z defd„ƒYZ dS(s Firefox 3 "cookies.sqlite" cookie persistence. Copyright 2008 John J Lee This code is free software; you can redistribute it and/or modify it under the terms of the BSD or ZPL 2.1 licenses (see the file COPYING.txt included with the distribution). iÿÿÿÿN(t CookieJartCookietMappingIterator(t isstringliket experimentalsmechanize.cookiestFirefox3CookieJarcBs¡eZdZedd„Zd„Zd„Zd„Zdd„Z dd„Z d„Z d„Z dddd „Z d „Zd „Zd „Zd „Zd„ZRS(sˆFirefox 3 cookie jar. The cookies are stored in Firefox 3's "cookies.sqlite" format. Constructor arguments: filename: filename of cookies.sqlite (typically found at the top level of a firefox profile directory) autoconnect: as a convenience, connect to the SQLite cookies database at Firefox3CookieJar construction time (default True) policy: an object satisfying the mechanize.CookiePolicy interface Note that this is NOT a FileCookieJar, and there are no .load(), .save() or .restore() methods. The database is in sync with the cookiejar object's state after each public method call. Following Firefox's own behaviour, session cookies are never saved to the database. The file is created, and an sqlite database written to it, if it does not already exist. The moz_cookies database table is created if it does not already exist. cCsktdƒtj||ƒ|dk rBt|ƒ rBtdƒ‚n||_d|_|rg|jƒndS(Ns&Firefox3CookieJar is experimental codesfilename must be string-like( RRt__init__tNoneRt ValueErrortfilenamet_conntconnect(tselfR t autoconnecttpolicy((sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyR1s   cCs;ddl}|j|jƒ|_d|j_|jƒdS(NiÿÿÿÿtDEFERRED(tsqlite3R R R tisolation_levelt_create_table_if_necessary(R R((sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyR ;s  cCs|jjƒdS(N(R tclose(R ((sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyRAscCs\y1|jjƒ}z||ƒ}Wd|jƒXWn|jjƒ‚nX|jjƒ|S(N(R tcursorRtrollbacktcommit(R tfunctcurtresult((sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyt _transactionDs  cs|j‡‡fd†ƒS(Ncs|jˆˆƒS(N(texecute(R(tquerytparams(sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pytSs(R(R RR((RRsA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyt_executeRscCs?|jjƒ}z|j||ƒ|jƒSWd|jƒXdS(N(R RRtfetchallR(R RRR((sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyt_queryUs cCs|jdƒdS(Ns¿CREATE TABLE IF NOT EXISTS moz_cookies (id INTEGER PRIMARY KEY, name TEXT, value TEXT, host TEXT, path TEXT,expiry INTEGER, lastAccessed INTEGER, isSecure INTEGER, isHttpOnly INTEGER)(R(R ((sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyR^scCs|\ }}}}}}}} } d} |jddƒ}|jddƒ}|jddƒ}|jddƒ}t| ƒ} i} | r”d| dR=sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyR?‹s"         c Csû|j}|jrd}nt|jƒ}t|jƒ}t|jƒ}t|jƒ}tt|j ƒƒ}|dkr‡|}d}ntt j ƒƒ} |j dƒ} |j dƒ} | jƒd} | dkrÜd} n| |||||| || f S(NR%R$s#SELECT MAX(id) + 1 from moz_cookiesii(R2R:tunicodeR0R1R.R/R(tintR4Rttimethas_nonstandard_attrRtfetchone( R tcookieRR2R0R1R.R/R4R3R5RR-((sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyt_row_from_cookie s&       cs@ˆjrtjˆˆƒdS‡‡fd†}ˆj|ƒdS(NcsXˆjˆ|ƒ}|dd!\}}}}|jd|||fƒ|jd|ƒdS(Niis@DELETE FROM moz_cookies WHERE host = ? AND path = ? AND name = ?s;INSERT INTO moz_cookies VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?) (RIR(RR,R.tunusedR0R1(R RH(sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyt set_cookie¿s(R:RRKR(R RHRK((R RHsA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyRKºs   ccsKxt|jƒD] }|VqWx%|jdƒD]}|j|ƒVq/WdS(Ns3SELECT * FROM moz_cookies ORDER BY name, path, host(Rt_cookiesR!R;(R RHR,((sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyt__iter__Ís    cs;tjˆˆƒ}‡‡fd†}ˆj|ƒ}||S(Ncsf|jdƒ}g|jƒD]}|d^q}g}x'|D]}|ˆj|ˆ|ƒ7}q?W|S(NsSELECT host from moz_cookiesi(RR t_persistent_cookies_for_domain(RRR,tdomainstcookiesR0(R trequest(sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyt get_cookiesØs#  (Rt_cookies_for_requestR(R RQtsession_cookiesRRtpersistent_coookies((R RQsA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyRSÖsc Csýg}|jj||ƒsgStd|ƒ|jd|fƒ}g|jƒD]}|j|ƒ^qN}d}g}x|D]y} | j|kr¹|jj| j|ƒ r¹| j}q|n|jj | |ƒsÞtdƒq|ntdƒ|j | ƒq|W|S(Ns!Checking %s for cookies to returns6SELECT * from moz_cookies WHERE host = ? ORDER BY paths not returning cookies it's a match( t_policytdomain_return_oktdebugRR R;RR1tpath_return_okt return_okR@( R R0RQRRPRR,t last_pathtrRH((sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyRNãs*  (    N(((t__name__t __module__t__doc__R+RRR RRRR!RR;R?RIRKRMRSRN(((sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyRs      '   ( R_tloggingREt _clientcookieRRRt_utilRRt getLoggerRXR(((sA/scratch/rashmi/Condor_Script/src/mechanize/_firefox3cookiejar.pyt s