mirror of
https://github.com/Lost-MSth/Arcaea-server.git
synced 2026-02-04 21:47:28 +08:00
32 lines
816 B
Python
32 lines
816 B
Python
import sqlite3
|
||
from flask import current_app
|
||
import traceback
|
||
|
||
|
||
class Connect():
|
||
# 数据库连接类，上下文管理
|
||
|
||
def __init__(self, file_path='./database/arcaea_database.db'):
|
||
"""
|
||
数据库连接，默认连接arcaea_database.db
|
||
接受：文件路径
|
||
返回：sqlite3连接操作对象
|
||
"""
|
||
self.file_path = file_path
|
||
|
||
def __enter__(self):
|
||
self.conn = sqlite3.connect(self.file_path)
|
||
self.c = self.conn.cursor()
|
||
return self.c
|
||
|
||
def __exit__(self, exc_type, exc_val, exc_tb):
|
||
if self.conn:
|
||
self.conn.commit()
|
||
self.conn.close()
|
||
|
||
if exc_type is not None:
|
||
current_app.logger.error(
|
||
traceback.format_exception(exc_type, exc_val, exc_tb))
|
||
|
||
return True
|