Archives/wikiteam
2
0
Fork 0
mirror of https://github.com/WikiTeam/wikiteam synced 2024-11-12 07:12:41 +00:00
Code Issues Projects Releases Wiki Activity

Merge branch 'PiRSquared17-follow-redirects-api'

Browse Source
This commit is contained in:
Federico Leva 2015-03-08 16:31:32 +01:00
commit c1a5e3e0ca
1 changed files with 30 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
dumpgenerator.py
View File

@ -1265,10 +1265,12 @@ def getParameters(params=[]):
# print index
index2 = None
check = api and checkAPI(api=api, session=session)
if check:
if api:
check = checkAPI(api=api, session=session)
if check[0]:
index2 = check[1]
print 'API is OK'
api = check[2]
print 'API is OK: ' + api
else:
print 'Error in API, please, provide a correct path to API'
sys.exit(1)
@ -1371,29 +1373,36 @@ def getParameters(params=[]):
def checkAPI(api=None, session=None):
""" Checking API availability """
global cj
r = session.post(
url=api,
data={
'action': 'query',
'meta': 'siteinfo',
'format': 'json'})
resultText = r.text
print 'Checking API...', api
# handle redirects
for i in range(4):
print 'Checking API...', api
r = session.post(
url=api,
data={
'action': 'query',
'meta': 'siteinfo',
'format': 'json'}
)
resultText = r.text
if r.url == api:
break
else:
api = r.url
if "MediaWiki API is not enabled for this site." in resultText:
return False
try:
result = json.loads(resultText)
if 'query' in result:
query = result['query']
general = result['query']['general']
if 'general' in query and 'script' in general and 'server' in general:
return (
True,
result['query']['general']['server'] +
result['query']['general']['script'])
else:
return (True, None)
index = None
if result['query']:
try:
index = result['query']['general']['server'] + \
result['query']['general']['script']
return ( True, index, api )
except ValueError:
print "MediaWiki API seems to work but returned no index URL"
return (True, None, api)
except ValueError:
print "MediaWiki API returned data we could not parse"
return False
return False