aboutsummaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
author简律纯 <i@jyunko.cn>2024-06-17 04:24:14 +0800
committer简律纯 <i@jyunko.cn>2024-06-17 04:24:14 +0800
commitdfec9f99d1726b3c3a4e9426f50c955fcaddb623 (patch)
tree6037a833b14cabd5ffda9b5a09f08229994fa616
parente914584e3e441a5b875e68d9443d99a745164e34 (diff)
downloadHydroRoll-dfec9f99d1726b3c3a4e9426f50c955fcaddb623.tar.gz
HydroRoll-dfec9f99d1726b3c3a4e9426f50c955fcaddb623.zip
chore(format): ruff lint code
-rw-r--r--examples/main.py2
-rw-r--r--hydro_roll/cli.py8
-rw-r--r--hydro_roll/config.py15
-rw-r--r--hydro_roll/models/utils.py3
-rw-r--r--hydro_roll/typing.py2
-rw-r--r--tests/test_sqlite3.py56
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()