paul@100 | 1 | # -*- coding: iso-8859-1 -*- |
paul@100 | 2 | """ |
paul@100 | 3 | getfiles - Obtain files using MoinMessage |
paul@100 | 4 | |
paul@100 | 5 | @copyright: 2013, 2014 by Paul Boddie <paul@boddie.org.uk> |
paul@100 | 6 | @license: GNU GPL (v2 or later), see COPYING.txt for details. |
paul@100 | 7 | """ |
paul@95 | 8 | |
paul@95 | 9 | from MoinMessage import * |
paul@95 | 10 | from email.mime.text import MIMEText |
paul@95 | 11 | from email.parser import Parser |
paul@95 | 12 | from os.path import join, exists |
paul@95 | 13 | from os import makedirs |
paul@112 | 14 | from time import strftime |
paul@95 | 15 | import sys |
paul@95 | 16 | |
paul@95 | 17 | def writefile(filename, s): |
paul@95 | 18 | f = open(filename, "wb") |
paul@95 | 19 | try: |
paul@95 | 20 | f.write(s) |
paul@95 | 21 | finally: |
paul@95 | 22 | f.close() |
paul@95 | 23 | |
paul@104 | 24 | def decrypt(gpg, content): |
paul@104 | 25 | if is_encrypted(content): |
paul@104 | 26 | text = gpg.decryptMessage(content) |
paul@104 | 27 | return Parser().parsestr(text) |
paul@104 | 28 | else: |
paul@104 | 29 | return content |
paul@104 | 30 | |
paul@114 | 31 | def save_part(gpg, directory, part): |
paul@114 | 32 | |
paul@114 | 33 | # The retrieved content may be encrypted. |
paul@114 | 34 | |
paul@114 | 35 | try: |
paul@114 | 36 | part = decrypt(gpg, part) |
paul@114 | 37 | except MoinMessageError: |
paul@114 | 38 | print >>sys.stderr, "Part could not be decrypted." |
paul@114 | 39 | return |
paul@114 | 40 | |
paul@114 | 41 | # The original sender may now be verified. |
paul@114 | 42 | |
paul@114 | 43 | if is_signed(part): |
paul@114 | 44 | fingerprint, identity, content = gpg.verifyMessage(part) |
paul@114 | 45 | print >>sys.stderr, "Content signed by %s." % identity |
paul@114 | 46 | else: |
paul@114 | 47 | content = part |
paul@114 | 48 | |
paul@114 | 49 | if content.is_multipart(): |
paul@114 | 50 | files = content.get_payload() |
paul@114 | 51 | else: |
paul@114 | 52 | files = [content] |
paul@114 | 53 | |
paul@114 | 54 | # Treat each part of the verified, possibly encrypted message as a file. |
paul@114 | 55 | |
paul@114 | 56 | for n, file in enumerate(files): |
paul@114 | 57 | if file.is_multipart(): |
paul@114 | 58 | save_part(gpg, directory, file) |
paul@114 | 59 | else: |
paul@114 | 60 | data = file.get_payload(decode=True) |
paul@114 | 61 | |
paul@114 | 62 | realname = file.get("Content-Disposition", str(n)) |
paul@114 | 63 | filename = join(directory, realname) |
paul@114 | 64 | |
paul@114 | 65 | dirpath = join(target_dir, directory) |
paul@114 | 66 | if not exists(dirpath): |
paul@114 | 67 | makedirs(dirpath) |
paul@114 | 68 | writefile(join(target_dir, filename), data) |
paul@114 | 69 | |
paul@95 | 70 | if __name__ == "__main__": |
paul@95 | 71 | try: |
paul@95 | 72 | service = sys.argv[1] |
paul@95 | 73 | signer = sys.argv[2] |
paul@95 | 74 | url = sys.argv[3] |
paul@95 | 75 | sender = sys.argv[4] |
paul@95 | 76 | datetime = sys.argv[5] |
paul@95 | 77 | target_dir = sys.argv[6] |
paul@95 | 78 | except IndexError: |
paul@95 | 79 | print >>sys.stderr, """\ |
paul@95 | 80 | Need a service key identifier, signing key identifier, the repository URL, |
paul@95 | 81 | sender identifier, the datetime of the last resources retrieved, and the target |
paul@95 | 82 | directory as arguments to this program. |
paul@95 | 83 | """ |
paul@95 | 84 | sys.exit(1) |
paul@95 | 85 | |
paul@95 | 86 | # Bundle files into a message. |
paul@95 | 87 | |
paul@95 | 88 | message = Message() |
paul@95 | 89 | |
paul@112 | 90 | commands = ["MBOX %s" % sender, "LAST %s" % datetime, "RETR"] |
paul@112 | 91 | part = MIMEText("\n".join(commands), "x-moinmessage-fetch") |
paul@95 | 92 | message.add_update(part) |
paul@95 | 93 | |
paul@95 | 94 | # Get the e-mail message itself. |
paul@95 | 95 | |
paul@104 | 96 | email_message = message.get_payload() |
paul@95 | 97 | |
paul@112 | 98 | # Sign, encrypt and send the request. |
paul@95 | 99 | |
paul@95 | 100 | gpg = GPG() |
paul@112 | 101 | signed_message = gpg.signMessage(email_message, signer) |
paul@112 | 102 | encrypted_message = gpg.encryptMessage(signed_message, service) |
paul@112 | 103 | resp = sendMessageOpener(encrypted_message, url) |
paul@95 | 104 | |
paul@104 | 105 | # Verify the response after possible transport encryption. |
paul@95 | 106 | |
paul@95 | 107 | try: |
paul@95 | 108 | message = Parser().parse(resp) |
paul@104 | 109 | message = decrypt(gpg, message) |
paul@95 | 110 | fingerprint, identity, content = gpg.verifyMessage(message) |
paul@95 | 111 | |
paul@95 | 112 | except MoinMessageDecodingError: |
paul@95 | 113 | print >>sys.stderr, "Incoming message was improperly encoded." |
paul@95 | 114 | sys.exit(1) |
paul@95 | 115 | |
paul@95 | 116 | except MoinMessageError, exc: |
paul@95 | 117 | print >>sys.stderr, "Incoming message was not verifiable: %s" % exc |
paul@95 | 118 | sys.exit(1) |
paul@95 | 119 | |
paul@95 | 120 | message = Message() |
paul@95 | 121 | message.handle_message(content) |
paul@95 | 122 | |
paul@95 | 123 | for part in message.updates: |
paul@112 | 124 | if part.get_content_type() == "text/x-moinmessage-fetch-result": |
paul@112 | 125 | if part["Request-Status"] != "OK": |
paul@112 | 126 | print part |
paul@112 | 127 | else: |
paul@112 | 128 | print part["Request-Type"] |
paul@114 | 129 | else: |
paul@114 | 130 | # Use the "outer" filename to determine a directory for the retrieved |
paul@114 | 131 | # file, even though the eventual filename in the directory may be |
paul@114 | 132 | # different. |
paul@95 | 133 | |
paul@114 | 134 | timestamp = strftime("%Y-%m-%d_%H:%M:%S") |
paul@114 | 135 | directory = part.get("Content-Disposition", timestamp) |
paul@114 | 136 | save_part(gpg, directory, part) |
paul@95 | 137 | |
paul@95 | 138 | # vim: tabstop=4 expandtab shiftwidth=4 |