Your IP : 216.73.216.85
K�V� � @ s� d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z e j d d k r� d d l Z e j
Z
d g Z d Z e
d � Z e e � d Z e j j e j j e j j e � � d � Z d d
� Z d d d
� Z d d � Z d d d � Z d d � Z d d d � Z d d � Z e e j j e d � � Z e e d � Z e e e d � � Z! Gd d � d e" � Z# d S) z�
captcha.audio
~~~~~~~~~~~~~
Generate Audio CAPTCHAs, with built-in digits CAPTCHA.
This module is totally inspired by https://github.com/dchest/captcha
� N� � AudioCaptchai@ s( RIFF WAVEfmt @ @ data� �datac C s2 t j | � } | j d � } | j � t | � S)N� ���)�wave�openZ
readframes�close� bytearray)�filepath�wr � r �>/opt/alt/python35/lib/python3.5/site-packages/captcha/audio.py�_read_wave_file# s
r r c C s� | d k r | St t | � | � } t | � } d } xa | D]Y } t | � } x: | t | | � k r� | | k r� | | | <| d 7} qT W| | 7} q? W| S)z(Change the voice speed of the wave body.r r )�int�lenr )�body�speed�length�rv�step�v�ir r r � change_speed* s
%
r c C s� t | � } | | d } t | } t j t � } t t j d | � � | d d � <| t t j d | � � 7} | | } | | k r� | t d g � } | S)zjPatch header to the given wave body.
:param body: the wave content body, it should be bytearray.
r z<Ir � r )r �WAVE_HEADER_LENGTH�copy� WAVE_HEADERr �struct�pack)r r Zpadded�total�headerr r r r �patch_wave_header< s
"
r# c C sk t | � } d t | d � } d } x> | | k rf t j d d � } | | | | | <| d 7} q) W| S)z!Create white noise for background� r r � r )r r �random�randint)r �level�noiseZadjustr r r r r � create_noiseT s r* c C s= t | � } d } x$ | | k r8 d | | <| d 7} q W| S)zCreate a piece of silence.r r$ r )r )r r r r r r �create_silence` s
r+ c C s� | d k r | St j | � } x� t | � D]� \ } } | d k r} | d | d } t t | � d � } t | d � } nB | d k r� d d | | } t t | � d � } t | d � } | | | <q, W| S)Nr r$ r% r )r � enumerate�maxr �min)r r( r r r r r � change_soundj s r/ c C s� t | � t | � k r% | | } } x t | � D]q \ } } | | } | d k r{ | d k r{ t | | d � | | <q2 t d | | | | d d � | | <q2 W| S)zMix two wave body into one.r$ r r% )r r, r )�src�dstr �svZdvr r r �mix_wave| s
,r3 zbeep.wavgffffff�?� c @ s� e Z d Z d Z d d d � Z e d d � � Z d d d � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d S)r a Create an audio CAPTCHA.
Create an instance of AudioCaptcha is pretty simple::
captcha = AudioCaptcha()
captcha.write('1234', 'out.wav')
This module has a built-in digits CAPTCHA, but it is suggested that you
create your own voice data library. A voice data library is a directory
that contains lots of single charater named directories, for example::
voices/
0/
1/
2/
The single charater named directories contain the wave files which pronunce
the directory name. A charater directory can has many wave files, this
AudioCaptcha will randomly choose one of them.
You should always use your own voice library::
captcha = AudioCaptcha(voicedir='/path/to/voices')
Nc C s1 | d k r t } | | _ i | _ g | _ d S)N)�DATA_DIR� _voicedir�_cache�_choices)�selfZvoicedirr r r �__init__� s
zAudioCaptcha.__init__c C sz | j r | j Sx` t j | j � D]L } t | � d k r# t j j t j j | j | � � r# | j j | � q# W| j S)z1Available choices for characters to be generated.r ) r8 �os�listdirr6 r �path�isdir�join�append)r9 �nr r r �choices� s 6zAudioCaptcha.choices� c C s t j | j | � S)zjGenerate a random string with the given length.
:param length: the return string length.
)r& �samplerB )r9 r r r r r& � s zAudioCaptcha.randomc C s% x | j D] } | j | � q
Wd S)z Load voice data into memory.N)rB �
_load_data)r9 �namer r r �load� s zAudioCaptcha.loadc C s� t j j | j | � } g } x` t j | � D]O } t j j | | � } | j d � r. t j j | � r. | j t | � � q. W| | j | <d S)Nz.wav)
r; r= r? r6 r< �endswith�isfiler@ r r7 )r9 rF �dirnamer �fr r r r rE � s !zAudioCaptcha._load_datac C sd t j | j | � } t j d d � d } t | | � } t j d d � d } t | | � } | S)N�Z �x g Y@�P )r& �choicer7 � randranger r/ )r9 �key�voicer r( r r r � _twist_pick� s zAudioCaptcha._twist_pickc C s� t j | j � } t j | j | � } t j | � } | j � t j d d � d } t | | � } t j d d � d } t | | � } | S)Nr � g $@r rC ) r& rO rB r7 r �reverserP r r/ )r9 rQ rR r r( r r r � _noise_pick� s
zAudioCaptcha._noise_pickc C s� t | d � } d } xs | | k r� | j � } | t | � d } t | | | | � � | | | � <| t j d t t d � � } q W| S)Nr r r �
)r* rV r r3 r& r' r �WAVE_SAMPLE_RATE)r9 r �charsr) �posZsound�endr r r �create_background_noise� s #$z$AudioCaptcha.create_background_noisec C s+ g } g } xG | D]? } | j | j | � � t j t t d � } | j | � q Wt d d � | � } t | � t | � t t j
| � } | j | | � } | d } x_ t | � D]Q \ }
} | t | � d } t
| | | | � � | | | � <| | |
} q� Wt t t t t | t S)N� c S s
t | � S)N)r )�ar r r �<lambda>� s z/AudioCaptcha.create_wave_body.<locals>.<lambda>r r )r@ rS r& r' rX �mapr- r �reduce�operator�addr\ r, r3 �BEEP�SILENCE�END_BEEP) r9 rY ZvoicesZintersrQ r Z durationsr �bgrZ r r[ r r r �create_wave_body� s
&
#z AudioCaptcha.create_wave_bodyc C s, | j s | j � | j | � } t | � S)zrGenerate audio CAPTCHA data. The return data is a bytearray.
:param chars: text to be generated.
)r7 rG rh r# )r9 rY r r r r �generate s
zAudioCaptcha.generatec C s9 | j | � } t | d � � } | j | � SWd QRXd S)z�Generate and write audio CAPTCHA data to the output.
:param chars: text to be generated.
:param output: output destionation.
�wbN)ri r �write)r9 rY �outputr rK r r r rk s zAudioCaptcha.write)�__name__�
__module__� __qualname__�__doc__r: �propertyrB r& rG rE rS rV r\ rh ri rk r r r r r � s
)$rp r; r r r r& rb �sys� version_info� functoolsra �__all__rX r r r r r= r? �abspathrJ �__file__r5 r r r# r* r+ r/ r3 rd rf r re �objectr r r r r �<module> s6 -