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

    В статье «Почему форматы 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")
    
  • Запись непойманных исключений в файл

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

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

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

    screenshot

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

    Подключить обработчик очень легко:

    import os
    import cgitb
    cgitb.enable(display=False, logdir=os.path.dirname(__file__))
    

    logdir — директория, куда будет записан html-файл с информацией об ошибке. В данном примере указана директория, в которой расположен исполняемый скрипт. В CGI-скриптах должна быть указана директория, к которой нет публичного доступа.

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

  • Работа с файлами 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. Полученный файл пишется на диск.

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

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

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

    Я остановил свой выбор на 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 сообщений.

Страница 49 из 49