Ivan Grishaev's blog

Создание файлов Excel

Aug 9, 2012 python, excel, programming
В статье «Почему форматы Microsoft Office такие сложные?» Джоэл Спольски пишет:

Заставьте Office делать грязную работу за вас. Word и Excel завязаны на сложных объектных моделях, основанных на автоматизации COM, что позволяет программно делать всё, что угодно. Во многих ситуациях проще будет использовать код Office, чем пытаться реализовать его с нуля.

Действительно, если требуется сформировать сложный офисный документ, проще сделать это вызовами API через COM-соединение. Однако, для создания простой эксель-таблички можно задействовать библиотеку xlwt.

Библиотека ставиться просто:
```
pip install xlwt
```
Требуется сформировать таблицу с тремя колонками:
1. номер лицевого счета абонента, строковый тип;
2. дата показания, тип дата;
3. показание, тип число.
Проект выполнен на Джанго. Есть модель показания, переданного абонентом:
```
class Statement(models.Model):
    user = models.ForeignKey(User)
    value = models.DecimalField(max_digits=12, decimal_places=6)
    date = models.DateTimeField(default=datetime.now)
```
Показания, которые необходимо записать в таблицу, извлекаются следующим образом
```
statements = Statement.objects.filter(date__range=(t1, t2))
```
Формирование файла Excel будет выглядеть так:
```
import xlwt

# Определяем формат ячеек для дат
date_style = xlwt.XFStyle()
date_style.num_format_str = "M/D/YY"

# Определяем формат ячеек для чисел
num_style = xlwt.XFStyle()
num_style.num_format_str = "0.00"

wbk = xlwt.Workbook() # Новая книга
sheet = wbk.add_sheet("statements") # Новый лист в книге

# Добавляем шапку таблицы
sheet.write(0, 0, u"Лицевой счет")
sheet.write(0, 1, u"Дата")
sheet.write(0, 2, u"Показание")

# Обход показаний, statements — объект QuerySet из Джанго
for i, s in enumerate(statements):

    # i — индекс итерации, используется как номер строки
    sheet.write(i+1, 0, s.user.username)
    sheet.write(i+1, 1, s.date, date_style)
    sheet.write(i+1, 2, s.value, num_style)

# Задаем ширину колонок
sheet.col(0).width = 5000
sheet.col(1).width = 5000
sheet.col(2).width = 5000

# Запись файла
wbk.save("path/to/file.xls")
```
Запись непойманных исключений в файл

Aug 9, 2012 programming, python, debug
На работе у меня много скриптов, написанных на питоне, и большая часть из них выполняется ночью. Проблема — любой, даже самый надежный скрипт может упасть. Не удалось соединиться с БД, не найден нужный файл, отвалился сетевой ресурс, прислали кривой реестр.

Обрабатывать исключения на каждой операции — утомительно, код распухает на глазах. Гораздо важнее не отлавливать каждую ошибку, а фиксировать факт ее появления и ситуацию, в которой она возникла.

Модуль cgitb устанавливает свой обработчик на всплывающее исключение. Он пишет в файл всю необходимую информацию: дату, время, исключение, его параметры, имя скрипта, фрагмент кода, где было возбуждено исключение и локальные переменные.

Имея эти данные, нетрудно выявить причину ошибки и внести в скрипт нужные коррективы. Модуль очень полезен для CGI-скриптов: ошибки не придется искать в логах Апача, тем более, что такой подробной информации вы там не найдете.

Подключить обработчик очень легко:
```
import os
import cgitb
cgitb.enable(display=False, logdir=os.path.dirname(__file__))
```
logdir — директория, куда будет записан html-файл с информацией об ошибке. В данном примере указана директория, в которой расположен исполняемый скрипт. В CGI-скриптах должна быть указана директория, к которой нет публичного доступа.

При работе с cgitb важно подключать его как можно быстрее, желательно, в начале скрипта. Это поможет обработать ошибки импорта библиотек.
Работа с файлами mailbox

Aug 9, 2012 python, programming, mailbox
Чем хорош почтовый клиент Mozilla Thunderbird?

Хотя бы тем, что хранит письма в удобном формате mailbox. Благодрая этому можно рулить почтой программно: считывать содержимое ящика, удалять и добавлять письма.

Зачем это может понадобитсья? Например, каждый день вам на почту приходят реестры платежей, которые нужно выгрузить из аттача, распаковать, и, в зависимости от имени файла, скопировать в определенную директорию. Напрашивается автоматизация процесса.

В составе богатой библиотеки Питона есть модуль mailbox. Небольшой пример:
```
import os
import cgitb
cgitb.enable(display=False, logdir=os.path.dirname(__file__))

import mailbox

MAILBOX_PATH = "path/to/mailbox.file" # Путь к файлу ящика
DATA_PATH = "store/path" # Директория для файлов

mbox = mailbox.mbox(MAILBOX_PATH, create=False)
for i, message in enumerate(mbox):
    attachments = message.get_payload()
    for attachment in attachments:
        filename = attachment.get_filename()
        if filename and filename.endswith(".txt"):
            body = attachment.get_payload(decode=True)
            filepath = os.path.join(DATA_PATH, filename)
            with open(filepath, "w") as f:
                f.write(body)
    mbox.remove(i)
mbox.close()
```
Основной цикл обходит все письма в ящике. Переменная attachments — это список частей письма. Частью может быть всё — простой текст, HTML-сообщение, файловое вложение и т.д.

get_filename() возвращает имя части. Для файлов оно будет отлично от None. Нас интересуют только текстовые файлы. Тело файла извлекается методом get_payload() части.

При передаче файла он может быть закодирован методами base64 или uuencode. Чтобы получить исходный файл, передается параметр decode=True. Полученный файл пишется на диск.

Сообщение удаляется из ящика. По окончании цикла ящик закрывается.

Рассылка смс в Питоне

Aug 8, 2012 sms, python, programming

Отправлять короткие сообщения можно с помощью любого из многочисленных сервисов смс-рассылок.

Я остановил свой выбор на LittleSMS. Низкая фиксированная цена, удобные API, подробная документация, готовые решения для многих языков и платформ.

Работа с сервисом осуществляется по протоколу HTTP GET-запросами. При регистрации вы получите API-ключ. Этим ключом подписываются все запросы. Ключ не является паролем к учетной записи. Если ключ где-то засветился, немедленно смените его в личном кабинете сервиса.

Для Питона рекомендую использовать мою библиотеку с Гитхаба.

Пример работы:

import littlesms

# Инициализация класса
api = littlesms.Api("user", "API_key")

# Проверка баланса.
print api.balance()
>>> {u'status': u'success', u'balance': 0.5}

# Отправка сообщения.
print api.send(u"Hello, World!", "7xxxxxxxxxx")
>>> {
        u'count': 1,
        u'status': u'success',
        u'recipients': [u'7xxxxxxxxxx'],
        u'price': 0.5,
        u'parts': 1,
        u'test': 0,
        u'balance': 0.5,
        u'messages_id': [u'xxxxxx']
}

# Отправка сообщения нескольким адресатам с подменой поля «отправитель».
recipients = ("7xxxxxxxxx1", "7xxxxxxxxx2", "7xxxxxxxxx3")
print api.send(u"Hello, World!", recipients, sender="Anonym")
>>> {
        u'count': 1,
        u'status': u'success',
        u'recipients': [u'7xxxxxxxxx1', u'7xxxxxxxxx2', u'7xxxxxxxxx3'],
        u'price': 0.5,
        u'parts': 1,
        u'test': 0,
        u'balance': 0.5, u'messages_id': [u'xxxxxx1', u'xxxxxx2', u'xxxxxx3']
}

# Если компьютер расположен за прокси со сложной схемой авторизации.
PROXY = {
    "proxy": "172.27.86.8",
    "port": 3128,
    "user": "ivan",
    "passw": "secret"
}
opener = littlesms.curl_opener(**PROXY)
api = littlesms.Api("user", "API_key", opener=opener)

# Пример работы в облачной платформе App Engine.
opener = littlesms.gae_opener()
api = littlesms.Api("user", "API_key", opener=opener)

# Пример обработки исключения.
try:
    print api.send(u"Hello, World!", "7xxxxxxxxxx", sender="TooLongSender!!!111")
except littlesms.ApiError, e:
    print e
    >>> Error 7: incorrect sender

Библиотека активно используется больше года в проекте личного кабинета Читинской энергосбытовой компании. На текущий момент отправлено около 15000 сообщений.

Создание файлов Excel

Запись непойманных исключений в файл

Работа с файлами mailbox

Рассылка смс в Питоне

← Страница 61 из 61