Skip to content

Commit 5bcef0f

Browse files
committed
fix: qrcode generate error
log debug -> info wave data too short
1 parent 574a28e commit 5bcef0f

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

dglabv3/dglab.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
from dglabv3.music_to_wave import convert_audio_to_v3_protocol
1616
from dglabv3.wsmessage import WSMessage, WStype
1717

18-
logging.basicConfig(level=logging.DEBUG)
18+
logging.basicConfig(level=logging.INFO)
1919
logger = logging.getLogger("dglabv3")
2020

2121

@@ -136,7 +136,7 @@ def generate_qrcode(self) -> Optional[io.BytesIO]:
136136
qr.make(fit=True)
137137
img = qr.make_image(fill_color="black", back_color="white")
138138
saveimg = io.BytesIO()
139-
img.save(saveimg, format="PNG") # type: ignore
139+
img.save(saveimg)
140140
saveimg.seek(0)
141141
return saveimg
142142

@@ -280,7 +280,9 @@ async def send_wave_message(self, wave: list[list[list[int]]], time: int = 10, c
280280
elif channel == Channel.BOTH:
281281
channel_str = "BOTH"
282282

283-
def _create_wave_message(ch_str: str, wave, time: int) -> dict:
283+
def _create_wave_message(ch_str: str, wave: list[list[list[int]]], time: int) -> dict:
284+
if len(wave) <= 4: # 避免波型過小
285+
wave = wave * 2
284286
return {
285287
"type": MessageType.CLIENT_MSG,
286288
"channel": ch_str,

0 commit comments

Comments
 (0)