login_model/cookies_pool.py at master · DoOpen/login_model · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
# -*- coding: utf-8 -*-
# @Time    : 2019/7/27 21:18
# @Author  : Esbiya
# @Email   : 18829040039@163.com
# @File    : redies_db.py
# @Software: PyCharm

from utils import *


class RedisClient:

    def __init__(self, logger):
        self.redis_client = get_redis_client()
        self.logger = logger

    def load_cookies(self, site, username):
        """
        从数据库读取 cookies
        :return:
        """
        cookies = self.redis_client.hget('cookies:{}'.format(site), username)

        if cookies:
            try:
                self.logger.info('从数据库读取 Cookies ...')
                cookies = json.loads(cookies)
            except ValueError:
                cookies = json.loads(json.dumps(ast.literal_eval(cookies)))
            self.logger.info('读取到 Cookies ! ')
            return cookies
        else:
            self.logger.warning('未从数据库中读取到 Cookies !')
            return False

    def save_cookies(self, site, username, cookies):
        """
        将 cookies 写入到数据库中
        :return:
        """
        self.logger.info('将 cookies 保存到数据库中...')
        cookies_str = json.dumps(cookies)
        self.redis_client.hset('cookies:{}'.format(site), username, cookies_str)
        self.logger.info('保存完成!')