Powershell + Gmail + Imgur.com = ... part 2

[Hikedaya]

Небольшое дополнение к скрипту отправки картинок на Imgur, закрывающее несколько моментов.
1. Считается, что в папке $UploadFolder что-то есть. Но ведь это не всегда так, от ошибок никто не застрахован. Потому проверяем эту папку на наличие там чего-либо. Если нет ничего - просто завершаем работу:

#no files - no message
if (!$files) { 
    write-host "Nothing to send, exiting sсript"
    exit
}

2. После того, как скрипт работу завершит, уже загруженные файлы стоило бы удалить, чтоб не мешались. Для этого последний оператор $smtp.send() оборачиваем в следующую конструкцию:

#send message
try {$smtp.Send($msg)}
catch {
    write-host "Error encountered while sending message, terminating sсript"
    #clean up objects to release files
    foreach ($att in $msg.attachments) {$att.dispose()}
    $msg.Dispose()
    exit
}
finally {
    #clean up objects to release files
    foreach ($att in $msg.attachments) {$att.dispose()}
    $msg.Dispose()
    #remove files from upload folder
    $files | remove-item
}

Для чего такие сложности? Из-за того, что по завершении отправки среда Powershell не прекратит ссылаться на наши файлы, и эту связь нужно сначала разорвать. Отдельно для каждого вложения и затем для всего письма в целом. После того, как связи разорваны, можно удалить все отправленные файлы. Что и делается в блоке finally. Блок же catch нужен для того, чтобы отследить любую ошибку во время передачи файлов, перехватить ее, опять таки, разорвать связи с файлами и завершить скрипт, не удаляя сами файлы.

3. Предположим, что в папке $UploadFolder пользователь хранит не только те файлы, которые предназначены к отправке (всякое бывает, помним?). Музыка, тексты, еще что-то. Выберем из них только те, что можно отправлять. Это JPG,GIF, PNG.

$IncludeFiles=@("*.jpg","*.gif","*.png")
$files=get-childitem $Uploadfolder -include $Includefiles -recurse

Окончательный вариант скрипта после всех этих изменений - ниже.

#setting up variables
$UploadFolder="%INSERT IGNORE-NAME-OF-FOLDER-WITH-PICTURES%"
$IncludeFiles=@("*.jpg","*.gif","*.png")
$SMTPServer = "smtp.gmail.com"
$SMTPPort = "587"
$SMTPUsername = "[e-mail]%"
$SMTPPassword = "%INSERT IGNORE-PASSWORD%"
$MailFrom = "[e-mail]%"
$MailTo = "[e-mail]"
$MailSubject = "Imgur Upload via Powershell"
$MailBody = "Powershell-generated message with attachments"

#get files to attach
$files=get-childitem $Uploadfolder -include $IncludeFiles -recurse

#no files - no message
if (!$files) { 
    write-host "Nothing to send, exiting sсript"
    exit
}

#setting up smtp client
$smtp = New-Object System.Net.Mail.SmtpClient($SMTPServer, $SMTPPort)
$smtp.Credentials = New-Object System.Net.NetworkCredential($SMTPUsername, $SMTPPassword)
$smtp.EnableSSL = $true

#settung up message
$msg = new-object Net.Mail.MailMessage
$msg.To.add($MailTo)
$msg.From = $MailFrom
$msg.Subject = $MailSubject
$msg.Body = $MailBody
$msg.sender = $SMTPMailFrom

#attach files in upload folder
foreach ($file in $files) {
    $MailAttachment=new-object Net.Mail.Attachment($file.fullname)
    $MailAttachment.ContentType.MediaType = "image/$($file.extension.substring(1))"
    $msg.Attachments.Add($MailAttachment)
}

#send message
try {$smtp.Send($msg)}
catch {
    write-host "Error encountered while sending message, terminating sсript"
    #clean up objects to release files
    foreach ($att in $msg.attachments) {$att.dispose()}
    $msg.Dispose()
    exit
}
finally {
    #clean up objects to release files
    foreach ($att in $msg.attachments) {$att.dispose()}
    $msg.Dispose()
    #remove files from upload folder
    $files | remove-item
}

@музыка: Stellardrone - Light Years

@темы: PowerShell