对Requests和sqlite3进行封装

封装了两个库,简化一下代码。第一个不是最终版,先用着,比如说,L30那里关于请求的参数,只列出了常用的几个。到时候有需求了再改。

ohRequests.py

这个封装会对请求失败的URL默认重新请求3次,避免网络波动导致抓取失败。

import requests
from faker import Faker


class ohRequests(object):
    def __init__(self, retries = 1):
        self.faker = Faker()
        self.GET = 'get'
        self.POST = 'post'
        self.RETRIES = retries

    def __request(self,
                method,
                url,
                headers = {},
                cookies = None,
                proxies = None,
                params  = None,
                data    = None,
                timeout = 5):

        # FIXME: 
        # 没有检测User-Agent, user-Agent之类的key名
        headers['user-agent'] = self.faker.user_agent()

        for i in range(self.RETRIES):
            req_methond = requests.post if method == 'post' else requests.get
            response = None
            try:
                response = req_methond(url, 
                                headers = headers,
                                cookies = cookies,
                                proxies = proxies,
                                data    = data,
                                timeout = timeout, 
                                params  = params
                            )
                response.encoding = 'utf-8'
                return response

            except Exception:
                print ('HTTPError, Status Code: {}'.format(response.status_code))

            print ("Requests Error. Retry[{}], max retries[{}]".format(i+1, self.RETRIES))

        return None

    def get(self, url, **kwargs):
        return self.__request('get', url, **kwargs)

    def post(self, url, **kwargs):
        return self.__request('post', url, **kwargs)

    def faker_user_agent(self):
        return self.faker.user_agent()

if __name__ == '__main__':
    r = ohRequests()
    req = r.get("http://www.baidu.com")
    print (req)

ohSqlite3.py

import sqlite3
import os

class ohSqlite3(object):
    """ APIs for using Sqlite3.

    Attributes:
        dbpath: the db file path for manipulation
        conn: connection object returned from sqlite3.connect()
        cursor: cursor object returned from conn

        SHOW_SQL_STATEMENT: whether print sql statement, default False
    """

    SHOW_SQL_STATEMENT = False

    def __init__(self, dbpath):
        """inits ohSqlite3 with dbpath"""
        self.dbpath = dbpath
        self.conn = None
        self.cursor = None

    def __enter__(self):
        self.connect()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.close()

    def connect(self):
        """init db connection
        
        Raises:
            sqlite3.Error: An error occurred when attempting to connect sqlite db.
        """
        try:
            self.conn = sqlite3.connect(self.dbpath)
        except sqlite3.Error as e:
            print ("Connect {} failed. Error: {}".foramt(self.dbpath, e.args[0]))
        
        self.cursor = self.conn.cursor()

    def execute(self, sql, args = None):
        """execute sql statement

        Args:
            sql: the sql statement to be executed
            args: a tuple of fields to be added to sql, default None

        Returns:
            result of sql statement execution

        Raises:
            Exception: any errors occurred when executing sql statement

        """
        if self.SHOW_SQL_STATEMENT:
            print ("[SQL]{}".format(sql))

        data = None
        try:
            if args:
                self.cursor.execute(sql, args)
            else:
                self.cursor.execute(sql)
            data = self.cursor.fetchall()
        except Exception as e:
            print (e, "[SQL]:" + sql.strip())
            self.conn.rollback()

        self.conn.commit()
        if data:
            return data
        return None

    def query(self, sql, args = None):
        """Same as execute(), different names"""
        return self.execute(sql, args = None)

    def close(self):
        """close all objects"""
        self.cursor.close()
        self.conn.close()

def unitest():
    testdb = "ohSqlite3_test.db"

    # test funcs
    db = ohSqlite3(testdb)
    db.connect()

    db.execute("CREATE TABLE stocks (date text, trans text, symbol text, qty real, price real)")

    db.execute("INSERT INTO stocks VALUES ('2006-01-05','BUY','RHAT',100,35.14)")
    db.execute("INSERT INTO stocks VALUES ('2006-03-28', 'BUY', 'IBM', 1000, 45.00)")
    db.execute("INSERT INTO stocks VALUES ('2006-04-05', 'BUY', 'MSFT', 1000, 72.00)")
    db.execute("INSERT INTO stocks VALUES ('2006-04-06', 'SELL', 'IBM', 500, 53.00)")

    assert 4 == db.execute("SELECT count(*) FROM stocks")[0][0]

    db.execute("DROP TABLE stocks")

    db.close()

    # test __with__ statement
    with ohSqlite3(testdb) as db:
        db.query("CREATE TABLE stocks (date text, trans text, symbol text, qty real, price real)")

        db.execute("INSERT INTO stocks VALUES ('2006-01-05','BUY','RHAT',100,35.14)")
        db.execute("INSERT INTO stocks VALUES ('2006-03-28', 'BUY', 'IBM', 1000, 45.00)")
        db.execute("INSERT INTO stocks VALUES ('2006-04-05', 'BUY', 'MSFT', 1000, 72.00)")
        db.execute("INSERT INTO stocks VALUES ('2006-04-06', 'SELL', 'IBM', 500, 53.00)")

        assert 4 == db.execute("SELECT count(*) FROM stocks")[0][0]

        db.execute("DROP TABLE stocks")

    os.remove(testdb)


if __name__ == "__main__":
    unitest()
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,658评论 6 496
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,482评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,213评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,395评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,487评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,523评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,525评论 3 414
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,300评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,753评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,048评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,223评论 1 343
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,905评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,541评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,168评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,417评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,094评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,088评论 2 352

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,646评论 18 139
  • 1、通过CocoaPods安装项目名称项目信息 AFNetworking网络请求组件 FMDB本地数据库组件 SD...
    阳明先生_X自主阅读 15,977评论 3 119
  • 奇怪的是,不知为什么,加我小的时候,我极力装得像大人;而在我已经不是小孩的时候,我又希望像个孩子。 ...
    虞静呀阅读 317评论 0 2
  • 今晚儿子做完作业,本想带他出去散步,儿子让我陪他做游戏,算了,明天再出去吧。 做完游戏赶紧让小家伙...
    宝钧阅读 142评论 0 6
  • 几点说明: 1.这是一部已经完结的小说,因此,如果没有极特殊情况,绝对不会弃坑不填。2.为网发时的清晰明了,我特意...
    叶初夏阅读 496评论 19 17