Skip to content

Commit 5436e45

Browse files
committed
style: wrap lines to 79 characters to comply with PEP8
1 parent 123b3b1 commit 5436e45

File tree

7 files changed

+51
-25
lines changed

7 files changed

+51
-25
lines changed

pyproject.toml

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,12 @@
11
[tool.flake8]
2-
max-line-length = 89
2+
max-line-length = 79
33
inline-quotes="double"
44
import-order-style="google"
5-
extend-ignore = [
6-
"R503",
7-
]
85
exclude = [
96
".git",
107
"__pycache__",
11-
"docs/source/conf.py",
128
"venv",
139
]
1410

1511
[tool.black]
16-
line-length = 89
12+
line-length = 79

tfinance/blast.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,8 @@
1111

1212
async def notify_assignees(context: CallbackContext):
1313
db = Database()
14-
# Перебираем всех пользователей и рассылаем каждому курсы их избранных акций.
14+
# Перебираем всех пользователей и
15+
# рассылаем каждому курсы их избранных акций.
1516
for user in db.get_users():
1617
if db.check_user_daily_notify(user):
1718
if user.favourite_stocks:

tfinance/database.py

Lines changed: 17 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,10 @@ class Database:
2525
def __init__(self):
2626
# Подключение к БД с отключенной проверкой потока.
2727
Path("sqlite").mkdir(exist_ok=True)
28-
self.con = sqlite3.connect(f"sqlite/{DATABASE_NAME}.db", check_same_thread=False)
28+
self.con = sqlite3.connect(
29+
f"sqlite/{DATABASE_NAME}.db",
30+
check_same_thread=False,
31+
)
2932
self.cur = self.con.cursor()
3033
self.setup()
3134

@@ -114,7 +117,8 @@ def add_prediction(self, user: User, stock_name: str, updown: str):
114117
if predictions[0]:
115118
prediction = f"{predictions[0]} {prediction}"
116119
self.cur.execute(
117-
f"UPDATE users SET prediction = ' {prediction}' WHERE id = {user.id}",
120+
f"UPDATE users SET prediction = ' {prediction}' "
121+
f"WHERE id = {user.id}",
118122
)
119123
self.con.commit()
120124

@@ -148,7 +152,9 @@ def delete_predictions(self, user: User):
148152
:param user: Экземпляр класса User с данными об этом пользователе.
149153
:return: None
150154
"""
151-
self.cur.execute(f"UPDATE users SET prediction = '' WHERE id = {user.id}")
155+
self.cur.execute(
156+
f"UPDATE users SET prediction = '' WHERE id = {user.id}",
157+
)
152158
self.con.commit()
153159

154160
def select_stock(self, user: User, stock_name: str):
@@ -164,7 +170,8 @@ def select_stock(self, user: User, stock_name: str):
164170
if selected_stocks:
165171
stock_name = f"{selected_stocks} {stock_name}"
166172
self.cur.execute(
167-
f"UPDATE users SET selected_stock = '{stock_name}' WHERE id = {user.id}",
173+
f"UPDATE users SET selected_stock = '{stock_name}' "
174+
f"WHERE id = {user.id}",
168175
)
169176
self.con.commit()
170177

@@ -257,7 +264,8 @@ def add_favourites_stocks(self, user: User, stock_name: str):
257264
if stocks and stocks[0]:
258265
stock_name = f"{stocks[0]} {stock_name}"
259266
self.cur.execute(
260-
f"UPDATE users SET favourites_stocks = '{stock_name}' WHERE id = {user.id}",
267+
f"UPDATE users SET favourites_stocks = '{stock_name}' "
268+
f"WHERE id = {user.id}",
261269
)
262270
self.con.commit()
263271

@@ -303,7 +311,8 @@ def remove_favourites_stock(self, user: User, stock_name: str):
303311
else:
304312
a = f"'{' '.join(a)}'"
305313
self.cur.execute(
306-
f"UPDATE users SET favourites_stocks = {a} WHERE id = {user.id}",
314+
f"UPDATE users SET favourites_stocks = {a} "
315+
f"WHERE id = {user.id}",
307316
)
308317
self.con.commit()
309318

@@ -314,7 +323,8 @@ def user_daily_notify(self, user: User):
314323
:return: None
315324
"""
316325
self.cur.execute(
317-
f"UPDATE users SET daily_notify = NOT daily_notify WHERE id = {user.id}",
326+
f"UPDATE users SET daily_notify = NOT daily_notify "
327+
f"WHERE id = {user.id}",
318328
)
319329
self.con.commit()
320330

tfinance/game.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,11 @@
22
from telegram.ext import ContextTypes, CallbackContext
33

44
from database import Database
5-
from exceptions import EmptyDataFrameError, PredictionAlreadySet, StockSelectedAlready
5+
from exceptions import (
6+
EmptyDataFrameError,
7+
PredictionAlreadySet,
8+
StockSelectedAlready,
9+
)
610
from functions import create_user
711
from graphics.visualize import check_stock_prices, do_stock_image
812
from models import User
@@ -17,12 +21,14 @@ async def game_menu(update: Update, context: ContextTypes.DEFAULT_TYPE):
1721
# Сохраняем id сообщения для возможности одновременной
1822
message_id = str(int(update.message.message_id) + 2)
1923
# Игры на многих акциях.
20-
# Прибавляем 2 т.к. отправляем 2 сообщения: фото и приписку к нему с клавиатурой.
24+
# Прибавляем 2 т.к. отправляем 2 сообщения:
25+
# фото и приписку к нему с клавиатурой.
2126
try:
2227
# Проверка на наличие аргументов.
2328
if not context.args:
2429
await update.message.reply_text(
25-
"Неправильно введена команда! Попробуйте: /game [индекс акции]",
30+
"Неправильно введена команда! "
31+
"Попробуйте: /game [индекс акции]",
2632
)
2733
# Проверка: была ли выбрана акция до этого? Избегаем читерства.
2834
if db.check_selected_stocks(user):
@@ -80,7 +86,8 @@ async def game_menu(update: Update, context: ContextTypes.DEFAULT_TYPE):
8086
)
8187
db.remove_selected_stock(user, message_id)
8288

83-
# Возвращаем 1, чтобы показать ConversationHandler'у состояние, в котором находимся.
89+
# Возвращаем 1, чтобы показать ConversationHandler'у состояние,
90+
# в котором находимся.
8491
return 1
8592

8693

tfinance/graphics/visualize.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
from exceptions import EmptyDataFrameError, WrongPeriodError
77

8-
time_periods = {
8+
TIME_PERIODS = {
99
"1d": "за 1 день",
1010
"5d": "за 5 дней",
1111
"1mo": "за 1 месяц",
@@ -27,7 +27,7 @@ def do_stock_image(stock_name, period="1mo"):
2727
:param period: 1d,5d,1mo,3mo,6mo,1y,2y,5y,10y,ytd,max
2828
:return: Картинка в байтовом формате.
2929
"""
30-
if period not in time_periods:
30+
if period not in TIME_PERIODS:
3131
raise WrongPeriodError
3232
# Забираем данные из Yahoo Finance.
3333
stock = yf.download(stock_name, period=period)
@@ -38,7 +38,10 @@ def do_stock_image(stock_name, period="1mo"):
3838

3939
# Создаем полотно с графиком.
4040
stock["Close"].plot(grid=True)
41-
plt.title(f"Курс акции {stock_name} {time_periods.get(period)}", fontsize=14)
41+
plt.title(
42+
f"Курс акции {stock_name} {TIME_PERIODS.get(period)}",
43+
fontsize=14,
44+
)
4245
plt.gca().set(ylabel="Price USD")
4346

4447
# Записываем полученный график в байты и возвращаем полученное изображение.

tfinance/main.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,9 @@
3535
level=logging.INFO,
3636
handlers=[
3737
logging.FileHandler(
38-
datetime.datetime.now(tz=TIMEZONE).strftime("logs/%Y-%m-%d_%H-%M-%S.log"),
38+
datetime.datetime.now(
39+
tz=TIMEZONE,
40+
).strftime("logs/%Y-%m-%d_%H-%M-%S.log"),
3941
encoding="utf-8",
4042
),
4143
logging.StreamHandler(),
@@ -84,7 +86,8 @@ async def get_stock_image(update: Update, context: ContextTypes.DEFAULT_TYPE):
8486
)
8587
except WrongPeriodError:
8688
await update.message.reply_text(
87-
"Неверный период. Доступные периоды: 1d,5d,1mo,3mo,6mo,1y,2y,5y,10y,ytd,max",
89+
"Неверный период. "
90+
"Доступные периоды: 1d,5d,1mo,3mo,6mo,1y,2y,5y,10y,ytd,max",
8891
)
8992
except EmptyDataFrameError:
9093
await update.message.reply_text(
@@ -150,7 +153,8 @@ async def follow(update: Update, context: ContextTypes.DEFAULT_TYPE):
150153
await update.message.reply_text("Акция не найдена")
151154
else:
152155
await update.message.reply_text(
153-
"Неверный способ ввода. /follow [индекс акции]. Например: /follow AAPL",
156+
"Неверный способ ввода. "
157+
"/follow [индекс акции]. Например: /follow AAPL",
154158
)
155159

156160

tfinance/stock.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,5 +53,10 @@ def get_all_stocks():
5353
.get("table")
5454
.get("rows")
5555
)
56-
stocks = [{"symbol": i.get("symbol"), "name": i.get("name")} for i in stocks]
56+
stocks = [
57+
{
58+
"symbol": i.get("symbol"),
59+
"name": i.get("name"),
60+
} for i in stocks
61+
]
5762
save_stocks("stocks.json", stocks)

0 commit comments

Comments
 (0)