Dictaphone/Запись голоса | Fsoky

import pyaudio
import wave

global CHUNK
global FORMAT
global CHANNELS
global RATE
global RECORD_SECONDS
global OUTPUT_FILENAME

CHUNK = 1024 # 1024 байта
FORMAT = pyaudio.paInt16 # 16 бит
CHANNELS = 2 # Каналы
RATE = 44100 # Частота кадров
RECORD_SECONDS = int( input( 'Время: ' ) ) # Время записи
OUTPUT_FILENAME = 'output.wav' # Название файла

def recordVoice():
    global p
    p = pyaudio.PyAudio() # Класс PyAudio

    global stream
    stream = p.open(
        format = FORMAT,
        channels = CHANNELS,
        rate = RATE,
        input = True,
        frames_per_buffer = CHUNK
     )

    print( '[!] Запись...' )

    global frames
    frames = []

    for i in range( 0, int( RATE / CHUNK * RECORD_SECONDS ) ):
        data = stream.read( CHUNK )
        frames.append( data )

    print( '[!] Запись завершена!' )

def completion():
    stream.stop_stream()
    stream.close()
    p.terminate()

def result():
    wf = wave.open( OUTPUT_FILENAME, 'wb' )
    wf.setnchannels( CHANNELS )
    wf.setsampwidth( p.get_sample_size( FORMAT ) ) #Возвращает размер (в байтах) для указанного значения формат образца
    wf.setframerate( RATE )
    wf.writeframes( b''.join( frames ) )
    wf.close()

def main():
    recordVoice()
    completion()
    result()

main()