alex/lichess-export
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Some pylint fixes

Browse Source
This commit is contained in:
Aleksey Lobanov
2017-07-15 20:16:02 +03:00
parent 09a3a243db
commit b570ce39a8
1 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
lichess.py
View File

@@ -5,7 +5,6 @@ import argparse
import os import os
import sys import sys
import shutil import shutil
import json
import time import time
import tempfile import tempfile
import logging import logging
@@ -25,6 +24,8 @@ REQUESTS_NEED_VERIFY = False
def getTempfileName(ext=""): def getTempfileName(ext=""):
"""Returns full path to temp file with with `ext` extension"""
return os.path.join( return os.path.join(
tempfile.gettempdir(), tempfile.gettempdir(),
sys.argv[0] + next(tempfile._get_candidate_names()) + ext sys.argv[0] + next(tempfile._get_candidate_names()) + ext
@@ -32,11 +33,15 @@ def getTempfileName(ext=""):
def getGamesListUrl(user_name, page, block_size=100): def getGamesListUrl(user_name, page, block_size=100):
"""Returns lichess url for downloading list of games"""
URL_TEMPLATE = "https://en.lichess.org/api/user/{}/games?page={}&nb={}" URL_TEMPLATE = "https://en.lichess.org/api/user/{}/games?page={}&nb={}"
return URL_TEMPLATE.format(user_name, page, block_size) return URL_TEMPLATE.format(user_name, page, block_size)
def getGamePgnUrl(game_id): def getGamePgnUrl(game_id):
"""Returns lichess url for game with `game_id` id"""
URL_TEMPLATE = "https://en.lichess.org/game/export/{}.pgn" URL_TEMPLATE = "https://en.lichess.org/game/export/{}.pgn"
return URL_TEMPLATE.format(game_id) return URL_TEMPLATE.format(game_id)
@@ -72,8 +77,8 @@ def getGamesList(user_name):
else: else:
cur_page_ind += 1 cur_page_ind += 1
logging.info("Downloaded {} game identifiers".format( logging.info("Downloaded {} game identifiers".format(
games_count) games_count
) ))
def downloadGamesToFile(game_ids, file_name, thread_count): def downloadGamesToFile(game_ids, file_name, thread_count):
@@ -93,11 +98,14 @@ def downloadGamesToFile(game_ids, file_name, thread_count):
logging.info("Downloaded {}".format(req.url)) logging.info("Downloaded {}".format(req.url))
file_pgn.write((req.text + "\n\n").encode("utf-8")) file_pgn.write((req.text + "\n\n").encode("utf-8"))
finally: finally:
[i.raw.release_conn() for i in req.history] for req_prev in req.history:
req_prev.release_conn()
req.raw.release_conn() req.raw.release_conn()
def writePgn(user_name, file_name, thread_count): def writePgn(user_name, file_name, thread_count):
"""Write pgn for selected usern to file"""
downloadGamesToFile(getGamesList(user_name), file_name, thread_count) downloadGamesToFile(getGamesList(user_name), file_name, thread_count)
@@ -139,14 +147,14 @@ if __name__ == "__main__":
) )
args = parser.parse_args() args = parser.parse_args()
logging_selector = { LOGGING_SELECTOR = {
"off": logging.ERROR, "off": logging.ERROR,
"info": logging.INFO, "info": logging.INFO,
"debug": logging.DEBUG "debug": logging.DEBUG
} }
logging.basicConfig( logging.basicConfig(
format="%(asctime)s %(levelname)-8s %(message)s", format="%(asctime)s %(levelname)-8s %(message)s",
level=logging_selector[args.logging], level=LOGGING_SELECTOR[args.logging],
datefmt="%H:%M:%S" datefmt="%H:%M:%S"
) )
args.name = args.name.lower() args.name = args.name.lower()