B
ˆê]¢@ ã @ s d dl Z d dlmZmZ d dlZd dlZd dlZd dlZd dlZd dl m
Z
d dl Z d dl m
Z
d dlZd dlZd dl mZ d dlmZmZmZ d dlmZ d dlmZmZ yd d lmZmZ d
ZW n e e fk
rà d ZY nX d dl Z d dl Z d dl Z d d l!m!Z! d d
l"m#Z# e#ƒ Z"e$e"j%ƒ d*dd„Z&e"j'd
fdd„Z(d+dd„Z)d,dd„Z*d-dd„Z+d.d d „Z,d/d d!„Z-d0d"d#„Z.d1d&d'„Z/d2d(d)„Z0dS )3é N)ÚKeyÚAttr)ÚMP3)Úsleep)Ústrftime)ÚImageÚ ImageDrawÚ ImageFont)Úlistdir)ÚisfileÚjoin)ÚLEDÚButtonFT)Údatetime)Ú
Configurationc C sL t d¡}| d¡ | r d}nd}| d¡ tjd| dd}| d ¡ dS )
Nz cerebro_utils.control_displayz&Entered the control_display method ...zvcgencmd display_power 1zvcgencmd display_power 0z#Calling the display control now ...z%sT)Úshellz Display control now complete!)ÚloggingÚ getLoggerÚinfoÚ
subprocessÚcall)Zenable_displayZcontrol_display_loggerZ display_cmdÚstatus© r úV/Users/sacholla/WorkDocs/My Projects/codecommit/project-cerebro/py_client/rpi_utils.pyÚcontrol_display\ s
r c C sj t d¡}| d¡ |r&| d¡ dS | d¡ d}d}dtj }tjd ||| |f dd
}| d ¡ dS ) Nzcerebro_utils.show_imagesz$Entered the switch_images method ...z'In Debug mode, so no showing of images!Tz,And now, starting up the slideshow again ...z ../scripts/shutoff-display.shz!../scripts/start-picture-frame.shz%s/picframe.start.logz%s && %s "%s" > %s 2>&1 &)r z$Switching of images is now complete!)r r r ÚconfigÚ__CEREBRO_LOGS_DIR__r r )Ú media_dirZ
debug_modeZshow_images_loggerZslideshow_stop_utilZslideshow_utilZ
slideshow_logr r r r Ú show_imagesn s
r ÚJoannaú sample textú
speech.mp3c C s t d¡}| d¡ t ¡ d¡}| d¡ | ¡ }t|ƒ}|dkr†t|d ƒ}d |d |… ¡}d ||d d … ¡} d || f }
n d
| d }
d t
j |f } t j
| ¡r¶td
ƒ nVtdƒ |j| dd|
d} | d¡ t| dƒ}
|
| d ¡ ¡ |
¡ | d| ¡ | S )Nz cerebro_utils.generate_audioz7Starting the generate_audio method in cerebro_utils ...Zpollyz*Now, going to synthesize_speech with pollyé
é ú é z7%s%szzz%s/%sz2
File already exists locally, not calling Polly..z&
File does not exist, making it now. Zmp3Zssml)ZVoiceIdZ OutputFormatZTextTypeÚTextz@Next, call to polly completed - now to stream the audio file ...ÚwbZ AudioStreamz-Finally, audio file is written and ready @ %s)r r r Úboto3ZSessionÚclientÚsplitÚlenÚintr r Z__CEREBRO_AUDIO_DIR__ÚosÚpathr ÚprintZsynthesize_speechÚopenÚwriteÚreadÚclose)Zvoice_idZ speech_textÚfilenameZgenerate_audio_loggerZ polly_clientZ
speech_tokensZ token_cntZ mid_tokenZpart_oneZpart_twoZspeech_text_with_bufferÚ file_pathÚresponseÚfiler r r Úgenerate_audio† s6
r8 Ú c C s¤ t d¡}| d¡ | d| ¡ | s4| d¡ dS |r@t|ƒ d}tjd|| f dd }|rt| d
| ¡ t|ƒ tj | ¡
¡ }|d kr–t | ¡ | d ¡ dS )
Nzcerebro_utils.play_audioz3Starting the play_audio method in cerebro_utils ...zFile_path provided : %szNo file_path provided!Fz sudo mpg321z%s %sT)r zSleeping for %d secsz
speech.mp3zdone here. audio played ?) r r r Úerrorr r r r- r. ÚbasenameÚstripÚremove)r5 Z
delay_secsZdelay_to_startZplay_audio_loggerZmp3_utilr r4 r r r Ú
play_audio s*
r> c C sŒ t d¡}| d¡ t d¡}| d| ¡ t| dƒ }|jd| ¡ idgd}W d Q R X | d ¡ | |¡ d
|kr„| d ¡ d S |d
S )
Nz(selfie_with_filters.get_facial_landmarksz&In the get_facial_landmarks method ...Z rekognitionz%Running Detect Faces on the image: %sÚrbÚBytesZALL)r Z
Attributesz#Completed the detect_faces API callZ FaceDetailszNo Faces found!F) r r r r( r) r0 Z detect_facesr2 r: )Ú
image_pathZget_facial_landmarks_loggerr) Úimager6 r r r Úget_facial_landmarksÜ s
"
rC ú&../assets/printing/reinvent_bkgrnd.jpgú#../assets/printing/reinventLogo.pngc C s* t d¡}| d¡ |s&| d¡ dS | s8| d¡ dS |sJ| d¡ dS | d¡ t |¡}t | ¡}t |¡}| |j¡ | |j¡ | |j¡ |jd d |jd
d f}| |tj¡ | d ¡}|jd d |jd
d f} | | tj¡ t
d
ƒt
dƒf}
| |
¡ | |
tj¡} | ¡ } t
| jd d ƒt
| jd
d ƒf}
| |
¡ |
||
¡ t dd¡}t | ¡}d}|
d d |
d
d f}|j|||dd |
d |
d
d f}| |¡ |
|||¡ t dd¡}t | ¡}d}|
d d |
d
d f}|j|||d d d tj }| |¡ |S ) Nz rpi_utils.generate_print_imagezIn generate_print_image ...z No background image providedFzNo image selectedzNo logo providedz All images provided correctly!r g ì?r% r" g Ð?g @@g t@gìQ¸ …ë¡?gš™™™™™¹?Z
Arial_Boldé0 z Cerebro: A Connected Photoboothéi éP )éÿ rI r )ÚfontÚfillÚVerdanaé zNov 16 2019 12:30:05 (PST)i é( )rI rI rI z%s/print_image.jpg)r r r r: r r0 ÚsizeÚ thumbnailÚ ANTIALIASÚrotater, ÚresizeÚcopyÚpaster Útruetyper ZDrawÚtextr Ú__CEREBRO_MEDIA_DIR__Úsave)Úimage_selectedZ bkgrnd_imageZ
logo_imageZgenerate_print_image_loggerÚimage1Úimage2Z imageLogoZ selfie_sizeZimage2_rotatedZ logo_sizeZbackground_sizeZbackground_imageZim3_copyZ image_boxZfntÚdZ title_stringZ title_boxZlogo_boxZfnt2Úd2Zgenerated_print_image_pathr r r Úgenerate_print_imageð s\
$
r_ c C s¶ t d¡}| d¡ t| d}| d| ¡ |} t ¡ }| ¡ }d}x2|D ]*}| ||| d ¡ tj| ¡ krJ|}qJW | d|| f ¡ | d| ¡ |
|| d i ¡ | d
¡ d S ) Nzrpi_utils.print_imagez!Attempting to run a print job ...)rZ zGenerated Print Image: %sr9 z
device-uriz8The Printer Chosen: %s , The Image Chosen is in path: %sz'Triggering the print job now for %s ...r$ zCompleted the print job!) r r r r_ ÚcupsÚ
ConnectionZ getPrintersr Z__PRINTER_TYPE__ÚlowerZ printFile)ÚfileNameZprint_image_loggerZgenerated_imageÚconnZprintersZ printer_nameÚprinterr r r Ú print_image6 s$
rf c C sÌ t d¡}| d¡ t d¡}| d¡ i }| |d< d|d< ttjƒ|d< |rVd |d
< d tjd œ}tj }| d
|t
|¡f ¡ t j
d|t
|¡|d}tj} d}
| dt
|
¡ ¡ ¡ t|
¡ ƒ} | d| ¡ xè|
¡ D ]Ü} | d| d ¡ | d }
| d }
|r d|tj |
¡f }ndtjtj |
¡f }| d|
¡ y| | ¡ |
|¡ W nP tjjk
rª } z,tdƒ |jd d dkr˜td ƒ n‚ W d d }~X Y nX | d | ¡ qÞW | d ¡ | S ) Nz cerebro_processor.download_mediazEntered Download_media ...Ús3zDownloading Media now ...ÚprofileÚyesÚaudioZimage_max_countÚ1Úignore_stock_profileszapplication/json)z Content-Typez x-api-keyzURL: %s, payload: %sÚPOST)ÚdataÚheadersr9 z Response: %sz Total Number of Media files: %dzProcessing Media: %s ...Z image_keyZ
image_captionz%s/%szTry downloading file: %sz Error seen!ÚErrorZCodeZ404zThe object does not exist.zImage Downloaded to %s.zMedia downloaded.) r r r r( ÚresourceÚstrr Z__IMAGE_MAX_COUNT__Z__APIGW_X_API_KEY__Z
__APIGW_API__ÚjsonÚdumpsÚrequestsÚrequestZ
__S3_BUCKET__Údebugr+ r- r. r; rX ZBucketZ
download_fileZbotocoreÚ
exceptionsZ ClientErrorr/ r6 )Z profile_namer rl Zdownload_media_loggerrg Úpayloadro Úurlr6 Z s3_bucketZs3_keyZ media_countÚitemZ
media_captionÚ
local_fileÚer r r Údownload_media] sT
r~ Ú setting_nameÚ
setting_valuec C sÜ t d¡}| d¡ | s&| d¡ dS d}t |¡}| ¡ }| d¡ d| }| d| ¡ | |¡ | ¡ d }t d
d d ƒ}t
¡ }t || ¡ ƒ} |
d ¡}
| || |
f} | d| ¡ | || ¡ | ¡ | d
¡ dS )Nz cerebro_utils.persist_settingzIn persist_setting ...zInvalid parameter provided!Fz
cerebro.dbz Created the DB connection...z6 DELETE from local_settings where setting_name = '%s' z'Now , executing sql: '%s', with params.zd INSERT INTO local_settings(setting_name,setting_value,last_updated,last_updated_dt) VALUES(?,?,?,?)i² r% z%m-%d-%YT%H:%M:%S.%fz!Setting is persisted into the db!T)r r r r: Úsqlite3ÚconnectÚcursorÚexecuteZcommitr Únowr, Ú
total_secondsr ) r r€ Zpersist_setting_loggerÚdb_filerd ÚcÚdml_sqlZ
epoch_timeZutc_timeZ last_updatedZlast_updated_dtZsettings_entryr r r Úpersist_settingœ s0
rŠ c C s´ t d¡}| d¡ | s&| d¡ dS d}t |¡}| ¡ }| d¡ d| }| d| ¡ | |¡ d }| ¡ }t |ƒd kršt |d
ƒd krš|d
d
}| d | ¡ |s°d }|S )
Nz cerebro_utils.retrieve_settingzIn retrieve_setting ...zInvalid parameter provided!Fz
cerebro.dbz Created the DB connection...zB SELECT setting_value FROM local_settings WHERE setting_name='%s' zNow , executing sql: '%s' .r% r z*Now , retrieved the setting value as: %s .r9 )
r r r r: r r‚ rƒ r„ Zfetchallr+ )r Zretrieve_setting_loggerr‡ rd rˆ r‰ r€ Úrowsr r r Úretrieve_settingÄ s*
rŒ )F)r r r! )r9 r r )r9 )r9 rD rE )r9 )r9 r9 F)r r€ )r )1r( Zboto3.dynamodb.conditionsr r r r r- Úglobrs Z mutagen.mp3r ÚmathÚtimer Úos.pathÚsysr ÚPILr r r r
r r Zgpiozeror
r Ztest_environmentÚ ImportErrorÚ RuntimeErrorr` ru r r r r r/ Ú__SQS_BACKEND_QUEUE__r rX r r8 r> rC r_ rf r~ rŠ rŒ r r r r Ú sN
4
'
/
D
'
?
(