python - 我可以将 cookie 从 python 导出到 Chrome 吗?

标签 python cookies sqlite python-requests pywin32

这就是我正在尝试做的事情。 我想从 Chrome 的 sqlite 文件中读取 cookie,在 python 上使用它们,然后再次将它们保存到 Chrome 的 sqlite 文件中,因为它们在与 python 一起使用时可能已被更新。

这或多或少是我必须测试的代码,我可以提取它们并正确使用它们,但我无法设法在 sqlite 上更新它们并继续在浏览器上使用它们。

我什至不确定我是否使用了正确的类型,或者是否将正确的类型传递给 UPDATE,在将 newEncrypted 传递给 sqlite 之前我是否应该对其执行任何操作?

  • 加密值的 type() 是缓冲区
  • 解密后的type()是unicode
  • newEncrypted 的 type() 是 str

    import shutil, cookielib, sqlite3, win32crypt
    
    def decrypt(result):
        cookies = []
    
        for host_key,name,value,expires_utc,encrypted_value in result:
            #print "Adding cookie"
            if value == '':
                print encrypted_value
                decrypted = win32crypt.CryptUnprotectData(encrypted_value, None, None, None, 0)[1].decode('utf-8')
            else:
                decrypted = value
    
    
            cookies.append(cookielib.Cookie(None, name, decrypted, '80', True, host_key, True, False, '/', True, False, (int(expires_utc)/1000000)-11644473600, False, None, None, None, False))
    
        return cookies
    
    cookie_file = "C:/Users/Daviid/AppData/Local/Google/Chrome/User Data/Default/Cookies"
    
    conn = sqlite3.connect(cookie_file)
    cursor = conn.cursor()
    
    sql = 'select host_key,name,value,expires_utc,encrypted_value from cookies where host_key = ".google.com" or host_key = "www.google.com"'
    cursor.execute(sql)
    
    result = cursor.fetchall()[0]
    cookies = decrypt(result)
    
    ###
    # Use Cookies in urllib, Requests, whatever...
    # cookies get updated
    ###
    updatedCookie = cookies[0]
    updatedCookie.value = "NewValue"
    
    newEncrypted = win32crypt.CryptProtectData(updatedCookie.value, '', None, None, None, 0)
    cursor.execute("UPDATE cookies SET encrypted_value=? WHERE host_key = ? AND name = ?",(newEncrypted, updatedCookie.domain, updatedCookie.name))
    
    cursor.close()
    conn.commit()
    conn.close()
    

最佳答案

sqlite 文档页面:[Python]: sqlite3 - DB-API 2.0 interface for SQLite databases .

code.py:

import sys
import sqlite3
import win32crypt


COOKIE_FILE = "ChromeCookies.db"
HOST_KEYS = ".google.com", "www.google.com"


def decrypt_cookies(query_result):
    cookies = []
    for host_key, name, value, expires_utc, encrypted in query_result:
        if encrypted:
            decrypted_descr, decrypted = win32crypt.CryptUnprotectData(encrypted, None, None, None, 0)
        else:
            decrypted_descr, decrypted = u"", u""
        cookies.append((host_key, name, expires_utc, decrypted, decrypted_descr, value))
        # Idx:              0       1        2           3            4            5
    return cookies


def init_db(file_name=COOKIE_FILE):
    conn = sqlite3.connect(file_name)
    return conn, conn.cursor()


def shutdown_db(cursor, connection, commit=False):
    cursor.close()
    if commit:
        connection.commit()
    connection.close()


def get_cookies(cursor, host_keys=HOST_KEYS):
    query_tmpl = "SELECT host_key, name, value, expires_utc, encrypted_value FROM cookies WHERE host_key IN {:}"
    cursor.execute(query_tmpl.format(host_keys))
    result = cursor.fetchall()
    cookies = decrypt_cookies(result)
    return cookies


def format_long_text(text, max_len=50):
    length = len(text)
    if length <= max_len:
        format_string = "({:d})[{:s}]"
    else:
        format_string =  "({:d})[{:s}...]"
    return format_string.format(length, text[:max_len])


def print_cookie(idx, cookie_tuple, print_data=False):
    print("\nIdx: {:d}\nHost key: {:s}\nName: {:s}\nExpires: {:d}".format(idx, *(cookie_tuple[:3])))
    if print_data:
         print("Decrypted: {:s}".format(format_long_text(cookie_tuple[3])))
         print("Decrypted description: {:s}".format(format_long_text(cookie_tuple[4])))
         print("Data: {:s}".format(format_long_text(cookie_tuple[5])))


def update_cookie(cursor, cookie_tuple):
    decrypted_length = len(cookie_tuple[3])
    encrypted = win32crypt.CryptProtectData(cookie_tuple[3][:decrypted_length // 2], cookie_tuple[4], None, None, None, 0)
    cursor.execute("UPDATE cookies SET encrypted_value = ? WHERE host_key = ? AND name = ?", (buffer(encrypted), cookie_tuple[0], cookie_tuple[1]))


def main():
    conn, cursor = init_db()
    cookies = get_cookies(cursor)
    cookie_idx = 8 % len(cookies) # For simplicity's sake choose an index that references a cookie with an unique `host_key` and `name`
    for idx, cookie_tuple in enumerate(cookies):
        print_cookie(idx, cookie_tuple)
    updated_cookie = cookies[cookie_idx]
    print_cookie(cookie_idx, updated_cookie, print_data=True)
    update_cookie(cursor, updated_cookie)
    shutdown_db(cursor, conn, commit=True)

    print("\nReopening DB...\n")

    conn, cursor = init_db()
    cookies = get_cookies(cursor)
    print("Cookies: {:d}".format(len(cookies)))
    #for idx, cookie_tuple in enumerate(cookies):
    #    print_cookie(idx, cookie_tuple)
    updated_cookie = cookies[cookie_idx]
    print_cookie(cookie_idx, updated_cookie, print_data=True)
    shutdown_db(cursor, conn)


if __name__ == "__main__":
    print("Python {:s} on {:s}\n".format(sys.version, sys.platform))
    main()

注释:

  • 我将代码拆分为函数,使其更加模块化(可重用和参数化)
  • 我没有安装cookielib,所以我删除了引用它的所有部分,并使用元组来存储cookie数据(我知道dictcollections.namedtuple 会更好,但我不想添加更多代码,因为这部分无论如何都不会保留)
  • 没有错误处理
  • 某些功能仅用于打印目的
  • 作为测试,我选择了其中一个 Cookie,并且仅保留其(加密)数据的前一半。当然它会失效,但这只是为了演示目的
  • key 是 buffer(encrypted)(在 update_cookie 中),因为 cookies.encrypted_value 列是一个 BLOB
  • 仅兼容Python2(注意到问题中的#print "Adding cookie"),因为在Python3中> 字符串是 unicode(并且 buffer 不存在)

输出:

(py27x64_test) e:\Work\Dev\StackOverflow\q050886719>"e:\Work\Dev\VEnvs\py27x64_test\Scripts\python.exe" code.py
Python 2.7.13 (v2.7.13:a06454b1afa1, Dec 17 2016, 20:53:40) [MSC v.1500 64 bit (AMD64)] on win32


Idx: 0
Host key: .google.com
Name: 1P_JAR
Expires: 13176376602000000

Idx: 1
Host key: .google.com
Name: AID
Expires: 13194604800101431

Idx: 2
Host key: .google.com
Name: APISID
Expires: 13236602265411651

Idx: 3
Host key: .google.com
Name: CGIC
Expires: 13189118241681530

Idx: 4
Host key: .google.com
Name: CGIC
Expires: 13189118241681619

Idx: 5
Host key: .google.com
Name: CONSENT
Expires: 13791196798240053

Idx: 6
Host key: .google.com
Name: GMAIL_RTT
Expires: 0

Idx: 7
Host key: .google.com
Name: HSID
Expires: 13236602265411610

Idx: 8
Host key: .google.com
Name: NID
Expires: 13189336603638418

Idx: 9
Host key: .google.com
Name: SAPISID
Expires: 13236602265411673

Idx: 10
Host key: .google.com
Name: SID
Expires: 13236602265411550

Idx: 11
Host key: .google.com
Name: SIDCC
Expires: 13181563951325551

Idx: 12
Host key: .google.com
Name: SNID
Expires: 13189336552795314

Idx: 13
Host key: .google.com
Name: SSID
Expires: 13236602265411631

Idx: 14
Host key: .google.com
Name: TAID
Expires: 13174995823101361

Idx: 15
Host key: www.google.com
Name: GAPS
Expires: 13229834000111924

Idx: 8
Host key: .google.com
Name: NID
Expires: 13189336603638418
Decrypted: (254)[132=tqLTXZ4dOSPbstdv6oktg9VxbNX3LpwLKGpXzpMwnyVTis...]
Decrypted description: (0)[]
Data: (0)[]

Reopening DB...

Cookies: 16

Idx: 8
Host key: .google.com
Name: NID
Expires: 13189336603638418
Decrypted: (127)[132=tqLTXZ4dOSPbstdv6oktg9VxbNX3LpwLKGpXzpMwnyVTis...]
Decrypted description: (0)[]
Data: (0)[]

关于python - 我可以将 cookie 从 python 导出到 Chrome 吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50886719/

相关文章:

iphone - 将本地 SQLite 数据库复制到 iPhone?

c# - 如何在 Sqlite 中使用 Dapper.QueryMultiple

python - Qt中两个MainWindow()之间的通信

node.js - Express.js : Cookie being saved late

javascript - 一个 cookie 上的多个值

javascript - 在网站上,是否可以填写不使用查询字符串参数的另一个域上的表单字段?

python - 将函数应用于数据框的元素

python - 如何从 pd.dataframe 获取 bool 值 'True' 的坐标?

python - 使用 matplotlib fill_between 在两条极坐标曲线之间填充

c# - SQLite :memory: database usage with using/dispose