diff options
| author | 2024-06-17 04:24:14 +0800 | |
|---|---|---|
| committer | 2024-06-17 04:24:14 +0800 | |
| commit | dfec9f99d1726b3c3a4e9426f50c955fcaddb623 (patch) | |
| tree | 6037a833b14cabd5ffda9b5a09f08229994fa616 | |
| parent | e914584e3e441a5b875e68d9443d99a745164e34 (diff) | |
| download | HydroRoll-dfec9f99d1726b3c3a4e9426f50c955fcaddb623.tar.gz HydroRoll-dfec9f99d1726b3c3a4e9426f50c955fcaddb623.zip | |
chore(format): ruff lint code
| -rw-r--r-- | examples/main.py | 2 | ||||
| -rw-r--r-- | hydro_roll/cli.py | 8 | ||||
| -rw-r--r-- | hydro_roll/config.py | 15 | ||||
| -rw-r--r-- | hydro_roll/models/utils.py | 3 | ||||
| -rw-r--r-- | hydro_roll/typing.py | 2 | ||||
| -rw-r--r-- | tests/test_sqlite3.py | 56 |
6 files changed, 52 insertions, 34 deletions
diff --git a/examples/main.py b/examples/main.py index 211b3d3..bb95e79 100644 --- a/examples/main.py +++ b/examples/main.py @@ -4,5 +4,3 @@ bot = Bot(hot_reload=True) if __name__ == "__main__": bot.run() - - diff --git a/hydro_roll/cli.py b/hydro_roll/cli.py index 1a36348..2588ad8 100644 --- a/hydro_roll/cli.py +++ b/hydro_roll/cli.py @@ -8,6 +8,7 @@ from typing import TYPE_CHECKING if TYPE_CHECKING: from .typing import * + class Atien(object): parser = argparse.ArgumentParser(description="水系终端脚手架") @@ -85,7 +86,12 @@ class Atien(object): print(f"下载包时出错:{package_name}") def build_template(self): - template = input("请选择应用模板(输入数字):\n" "1. 创建轻量应用\n" "2. 创建标准应用\n" "3. 创建开发应用\n") + template = input( + "请选择应用模板(输入数字):\n" + "1. 创建轻量应用\n" + "2. 创建标准应用\n" + "3. 创建开发应用\n" + ) if template == "1": print("选择了轻量应用模板") diff --git a/hydro_roll/config.py b/hydro_roll/config.py index e0789a0..a1b1d8a 100644 --- a/hydro_roll/config.py +++ b/hydro_roll/config.py @@ -83,15 +83,14 @@ class Directory(object): self.current_path = _path def get_dice_dir_list(self, _prefix: str) -> list: - return [ - os.path.join(self.current_path, f'{_prefix}', *dirs) + os.path.join(self.current_path, f"{_prefix}", *dirs) for dirs in [ - ['config'], - ['data'], - ['rules'], - ['scripts'], - ['web', 'frontend'], - ['web', 'backend'], + ["config"], + ["data"], + ["rules"], + ["scripts"], + ["web", "frontend"], + ["web", "backend"], ] ] diff --git a/hydro_roll/models/utils.py b/hydro_roll/models/utils.py index 73e7ba0..9e27307 100644 --- a/hydro_roll/models/utils.py +++ b/hydro_roll/models/utils.py @@ -1,6 +1,7 @@ import difflib import pickle + def find_max_similarity(input_string, string_list): max_similarity = 0 max_string = "" @@ -11,4 +12,4 @@ def find_max_similarity(input_string, string_list): max_similarity = similarity max_string = string - return max_string, max_similarity
\ No newline at end of file + return max_string, max_similarity diff --git a/hydro_roll/typing.py b/hydro_roll/typing.py index 876fa92..da83c36 100644 --- a/hydro_roll/typing.py +++ b/hydro_roll/typing.py @@ -31,4 +31,4 @@ T_Plugin = TypeVar("T_Plugin", bound="Plugin") T_Config = TypeVar("T_Config", bound="ConfigModel") T_BotHook = Callable[["Bot"], Awaitable[NoReturn]] -T_EventHook = Callable[[T_Event], Awaitable[NoReturn]]
\ No newline at end of file +T_EventHook = Callable[[T_Event], Awaitable[NoReturn]] diff --git a/tests/test_sqlite3.py b/tests/test_sqlite3.py index 92b0f59..9a7080c 100644 --- a/tests/test_sqlite3.py +++ b/tests/test_sqlite3.py @@ -1,13 +1,14 @@ import sqlite3 + class BottlePlugin: def __init__(self): - self.conn = sqlite3.connect('bottles.db') + self.conn = sqlite3.connect("bottles.db") self.cursor = self.conn.cursor() self.create_table() def create_table(self): - self.cursor.execute(''' + self.cursor.execute(""" CREATE TABLE IF NOT EXISTS bottles ( id INTEGER PRIMARY KEY AUTOINCREMENT, message TEXT NOT NULL, @@ -15,48 +16,63 @@ class BottlePlugin: receiver TEXT NOT NULL, status INTEGER DEFAULT 0 ); - ''') - self.cursor.execute(''' + """) + self.cursor.execute(""" CREATE TABLE IF NOT EXISTS comments ( id INTEGER PRIMARY KEY AUTOINCREMENT, bottle_id INTEGER NOT NULL, comment TEXT NOT NULL ); - ''') - self.cursor.execute(''' + """) + self.cursor.execute(""" CREATE TABLE IF NOT EXISTS likes ( id INTEGER PRIMARY KEY AUTOINCREMENT, bottle_id INTEGER NOT NULL, user TEXT NOT NULL ); - ''') + """) self.conn.commit() def throw_bottle(self, message, sender): - self.cursor.execute("INSERT INTO bottles (message, sender, receiver) VALUES (?, ?, '')", (message, sender)) + self.cursor.execute( + "INSERT INTO bottles (message, sender, receiver) VALUES (?, ?, '')", + (message, sender), + ) self.conn.commit() def pick_bottle(self, receiver): - self.cursor.execute("SELECT id, message, sender FROM bottles WHERE receiver='' OR receiver=? ORDER BY RANDOM() LIMIT 1", (receiver,)) + self.cursor.execute( + "SELECT id, message, sender FROM bottles WHERE receiver='' OR receiver=? ORDER BY RANDOM() LIMIT 1", + (receiver,), + ) row = self.cursor.fetchone() if row is None: return None bottle_id, message, sender = row - self.cursor.execute("UPDATE bottles SET receiver=? WHERE id=?", (receiver, bottle_id)) + self.cursor.execute( + "UPDATE bottles SET receiver=? WHERE id=?", (receiver, bottle_id) + ) self.conn.commit() - return {'id': bottle_id, 'message': message, 'sender': sender} + return {"id": bottle_id, "message": message, "sender": sender} def add_comment(self, bottle_id, comment): - self.cursor.execute("INSERT INTO comments (bottle_id, comment) VALUES (?, ?)", (bottle_id, comment)) + self.cursor.execute( + "INSERT INTO comments (bottle_id, comment) VALUES (?, ?)", + (bottle_id, comment), + ) self.conn.commit() def get_comments(self, bottle_id): - self.cursor.execute("SELECT comment FROM comments WHERE bottle_id=?", (bottle_id,)) + self.cursor.execute( + "SELECT comment FROM comments WHERE bottle_id=?", (bottle_id,) + ) rows = self.cursor.fetchall() return [row[0] for row in rows] def like_bottle(self, bottle_id, user): - self.cursor.execute("INSERT INTO likes (bottle_id, user) VALUES (?, ?)", (bottle_id, user)) + self.cursor.execute( + "INSERT INTO likes (bottle_id, user) VALUES (?, ?)", (bottle_id, user) + ) self.conn.commit() def get_likes(self, bottle_id): @@ -71,27 +87,25 @@ class BottlePlugin: self.conn.commit() - - plugin = BottlePlugin() # 扔一个漂流瓶 -plugin.throw_bottle('Hello, world!', 'Alice') +plugin.throw_bottle("Hello, world!", "Alice") # 捡一个漂流瓶 -bottle = plugin.pick_bottle('Bob') +bottle = plugin.pick_bottle("Bob") if bottle is not None: print(bottle) # 添加评论 -plugin.add_comment(1, 'Nice message!') +plugin.add_comment(1, "Nice message!") comments = plugin.get_comments(1) print(comments) # 点赞 -plugin.like_bottle(1, 'Charlie') +plugin.like_bottle(1, "Charlie") likes = plugin.get_likes(1) print(likes) # 清空漂流瓶 -# plugin.clear_bottles()
\ No newline at end of file +# plugin.clear_bottles() |
