Skip to content

Commit 55a46a6

Browse files
authoredAug 29, 2023
Merge pull request #5 from marksmayo/master
Pylint fixes binux#986
2 parents 0bd45f7 + f7634c5 commit 55a46a6

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

79 files changed

+384
-435
lines changed
 

‎docs/conf.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
class Mock(MagicMock):
1313
@classmethod
1414
def __getattr__(cls, name):
15-
return Mock()
15+
return Mock()
1616

1717
MOCK_MODULES = ['pycurl', 'lxml', 'psycopg2']
1818
sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)

‎pyspider/database/__init__.py

+32-45
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,9 @@
55
# http://binux.me
66
# Created on 2014-10-08 15:04:08
77

8-
import os, requests, json
8+
import os
9+
import json
10+
import requests
911
from six.moves.urllib.parse import urlparse, parse_qs
1012

1113

@@ -55,48 +57,38 @@ def _connect_database(url): # NOQA
5557
scheme = parsed.scheme.split('+')
5658
if len(scheme) == 1:
5759
raise Exception('wrong scheme format: %s' % parsed.scheme)
58-
else:
59-
engine, dbtype = scheme[0], scheme[-1]
60-
other_scheme = "+".join(scheme[1:-1])
60+
engine, dbtype = scheme[0], scheme[-1]
61+
other_scheme = "+".join(scheme[1:-1])
6162

6263
if dbtype not in ('taskdb', 'projectdb', 'resultdb'):
6364
raise LookupError('unknown database type: %s, '
6465
'type should be one of ["taskdb", "projectdb", "resultdb"]', dbtype)
6566

6667
if engine == 'mysql':
6768
return _connect_mysql(parsed,dbtype)
68-
69-
elif engine == 'sqlite':
69+
if engine == 'sqlite':
7070
return _connect_sqlite(parsed,dbtype)
71-
elif engine == 'mongodb':
71+
if engine == 'mongodb':
7272
return _connect_mongodb(parsed,dbtype,url)
73-
74-
elif engine == 'sqlalchemy':
73+
if engine == 'sqlalchemy':
7574
return _connect_sqlalchemy(parsed, dbtype, url, other_scheme)
76-
77-
78-
elif engine == 'redis':
75+
if engine == 'redis':
7976
if dbtype == 'taskdb':
8077
from .redis.taskdb import TaskDB
8178
return TaskDB(parsed.hostname, parsed.port,
8279
int(parsed.path.strip('/') or 0))
83-
else:
84-
raise LookupError('not supported dbtype: %s', dbtype)
85-
elif engine == 'local':
80+
raise LookupError('not supported dbtype: %s', dbtype)
81+
if engine == 'local':
8682
scripts = url.split('//', 1)[1].split(',')
8783
if dbtype == 'projectdb':
8884
from .local.projectdb import ProjectDB
8985
return ProjectDB(scripts)
90-
else:
91-
raise LookupError('not supported dbtype: %s', dbtype)
92-
elif engine == 'elasticsearch' or engine == 'es':
86+
raise LookupError('not supported dbtype: %s', dbtype)
87+
if engine == 'elasticsearch' or engine == 'es':
9388
return _connect_elasticsearch(parsed, dbtype)
94-
95-
elif engine == 'couchdb':
89+
if engine == 'couchdb':
9690
return _connect_couchdb(parsed, dbtype, url)
97-
98-
else:
99-
raise Exception('unknown engine: %s' % engine)
91+
raise Exception('unknown engine: %s' % engine)
10092

10193

10294
def _connect_mysql(parsed,dbtype):
@@ -115,14 +107,13 @@ def _connect_mysql(parsed,dbtype):
115107
if dbtype == 'taskdb':
116108
from .mysql.taskdb import TaskDB
117109
return TaskDB(**parames)
118-
elif dbtype == 'projectdb':
110+
if dbtype == 'projectdb':
119111
from .mysql.projectdb import ProjectDB
120112
return ProjectDB(**parames)
121-
elif dbtype == 'resultdb':
113+
if dbtype == 'resultdb':
122114
from .mysql.resultdb import ResultDB
123115
return ResultDB(**parames)
124-
else:
125-
raise LookupError
116+
raise LookupError
126117

127118

128119
def _connect_sqlite(parsed,dbtype):
@@ -138,14 +129,13 @@ def _connect_sqlite(parsed,dbtype):
138129
if dbtype == 'taskdb':
139130
from .sqlite.taskdb import TaskDB
140131
return TaskDB(path)
141-
elif dbtype == 'projectdb':
132+
if dbtype == 'projectdb':
142133
from .sqlite.projectdb import ProjectDB
143134
return ProjectDB(path)
144-
elif dbtype == 'resultdb':
135+
if dbtype == 'resultdb':
145136
from .sqlite.resultdb import ResultDB
146137
return ResultDB(path)
147-
else:
148-
raise LookupError
138+
raise LookupError
149139

150140

151141
def _connect_mongodb(parsed,dbtype,url):
@@ -157,14 +147,13 @@ def _connect_mongodb(parsed,dbtype,url):
157147
if dbtype == 'taskdb':
158148
from .mongodb.taskdb import TaskDB
159149
return TaskDB(url, **parames)
160-
elif dbtype == 'projectdb':
150+
if dbtype == 'projectdb':
161151
from .mongodb.projectdb import ProjectDB
162152
return ProjectDB(url, **parames)
163-
elif dbtype == 'resultdb':
153+
if dbtype == 'resultdb':
164154
from .mongodb.resultdb import ResultDB
165155
return ResultDB(url, **parames)
166-
else:
167-
raise LookupError
156+
raise LookupError
168157

169158

170159
def _connect_sqlalchemy(parsed, dbtype,url, other_scheme):
@@ -174,14 +163,13 @@ def _connect_sqlalchemy(parsed, dbtype,url, other_scheme):
174163
if dbtype == 'taskdb':
175164
from .sqlalchemy.taskdb import TaskDB
176165
return TaskDB(url)
177-
elif dbtype == 'projectdb':
166+
if dbtype == 'projectdb':
178167
from .sqlalchemy.projectdb import ProjectDB
179168
return ProjectDB(url)
180-
elif dbtype == 'resultdb':
169+
if dbtype == 'resultdb':
181170
from .sqlalchemy.resultdb import ResultDB
182171
return ResultDB(url)
183-
else:
184-
raise LookupError
172+
raise LookupError
185173

186174

187175
def _connect_elasticsearch(parsed, dbtype):
@@ -198,10 +186,10 @@ def _connect_elasticsearch(parsed, dbtype):
198186
if dbtype == 'projectdb':
199187
from .elasticsearch.projectdb import ProjectDB
200188
return ProjectDB([parsed.netloc], index=index)
201-
elif dbtype == 'resultdb':
189+
if dbtype == 'resultdb':
202190
from .elasticsearch.resultdb import ResultDB
203191
return ResultDB([parsed.netloc], index=index)
204-
elif dbtype == 'taskdb':
192+
if dbtype == 'taskdb':
205193
from .elasticsearch.taskdb import TaskDB
206194
return TaskDB([parsed.netloc], index=index)
207195

@@ -220,11 +208,10 @@ def _connect_couchdb(parsed, dbtype, url):
220208
if dbtype == 'taskdb':
221209
from .couchdb.taskdb import TaskDB
222210
return TaskDB(url, **params)
223-
elif dbtype == 'projectdb':
211+
if dbtype == 'projectdb':
224212
from .couchdb.projectdb import ProjectDB
225213
return ProjectDB(url, **params)
226-
elif dbtype == 'resultdb':
214+
if dbtype == 'resultdb':
227215
from .couchdb.resultdb import ResultDB
228216
return ResultDB(url, **params)
229-
else:
230-
raise LookupError
217+
raise LookupError

0 commit comments

Comments
 (0)
Please sign in to comment.