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@95 | 14 | import sys |
paul@95 | 15 | |
paul@95 | 16 | def writefile(filename, s): |
paul@95 | 17 | f = open(filename, "wb") |
paul@95 | 18 | try: |
paul@95 | 19 | f.write(s) |
paul@95 | 20 | finally: |
paul@95 | 21 | f.close() |
paul@95 | 22 | |
paul@104 | 23 | def decrypt(gpg, content): |
paul@104 | 24 | if is_encrypted(content): |
paul@104 | 25 | text = gpg.decryptMessage(content) |
paul@104 | 26 | return Parser().parsestr(text) |
paul@104 | 27 | else: |
paul@104 | 28 | return content |
paul@104 | 29 | |
paul@95 | 30 | if __name__ == "__main__": |
paul@95 | 31 | try: |
paul@95 | 32 | service = sys.argv[1] |
paul@95 | 33 | signer = sys.argv[2] |
paul@95 | 34 | url = sys.argv[3] |
paul@95 | 35 | sender = sys.argv[4] |
paul@95 | 36 | datetime = sys.argv[5] |
paul@95 | 37 | target_dir = sys.argv[6] |
paul@95 | 38 | except IndexError: |
paul@95 | 39 | print >>sys.stderr, """\ |
paul@95 | 40 | Need a service key identifier, signing key identifier, the repository URL, |
paul@95 | 41 | sender identifier, the datetime of the last resources retrieved, and the target |
paul@95 | 42 | directory as arguments to this program. |
paul@95 | 43 | """ |
paul@95 | 44 | sys.exit(1) |
paul@95 | 45 | |
paul@95 | 46 | # Bundle files into a message. |
paul@95 | 47 | |
paul@95 | 48 | message = Message() |
paul@95 | 49 | |
paul@95 | 50 | part = MIMEText("\n".join([sender, datetime]), "moinmessage-request") |
paul@95 | 51 | message.add_update(part) |
paul@95 | 52 | |
paul@95 | 53 | # Get the e-mail message itself. |
paul@95 | 54 | |
paul@104 | 55 | email_message = message.get_payload() |
paul@95 | 56 | |
paul@104 | 57 | # Encrypt, sign and send the request. |
paul@95 | 58 | |
paul@95 | 59 | gpg = GPG() |
paul@104 | 60 | encrypted_message = gpg.encryptMessage(email_message, service) |
paul@104 | 61 | signed_message = gpg.signMessage(encrypted_message, signer) |
paul@104 | 62 | resp = sendMessageForReading(signed_message, url) |
paul@95 | 63 | |
paul@104 | 64 | # Verify the response after possible transport encryption. |
paul@95 | 65 | |
paul@95 | 66 | try: |
paul@95 | 67 | message = Parser().parse(resp) |
paul@104 | 68 | message = decrypt(gpg, message) |
paul@95 | 69 | fingerprint, identity, content = gpg.verifyMessage(message) |
paul@95 | 70 | |
paul@95 | 71 | except MoinMessageDecodingError: |
paul@95 | 72 | print >>sys.stderr, "Incoming message was improperly encoded." |
paul@95 | 73 | sys.exit(1) |
paul@95 | 74 | |
paul@95 | 75 | except MoinMessageError, exc: |
paul@95 | 76 | print >>sys.stderr, "Incoming message was not verifiable: %s" % exc |
paul@95 | 77 | sys.exit(1) |
paul@95 | 78 | |
paul@95 | 79 | message = Message() |
paul@95 | 80 | message.handle_message(content) |
paul@95 | 81 | |
paul@95 | 82 | for part in message.updates: |
paul@95 | 83 | |
paul@95 | 84 | # Use the "outer" filename to determine a directory for the retrieved |
paul@95 | 85 | # file, even though the eventual filename in the directory may be |
paul@95 | 86 | # different. |
paul@95 | 87 | |
paul@104 | 88 | directory = part["Content-Disposition"] |
paul@95 | 89 | |
paul@104 | 90 | # The retrieved content may be encrypted. |
paul@104 | 91 | |
paul@104 | 92 | part = decrypt(gpg, part) |
paul@95 | 93 | |
paul@104 | 94 | # The original sender may now be verified. |
paul@95 | 95 | |
paul@104 | 96 | if is_signed(part): |
paul@104 | 97 | fingerprint, identity, content = gpg.verifyMessage(part) |
paul@104 | 98 | print >>sys.stderr, "Content signed by %s." % identity |
paul@95 | 99 | |
paul@95 | 100 | if content.is_multipart(): |
paul@95 | 101 | files = content.get_payload() |
paul@95 | 102 | else: |
paul@95 | 103 | files = [content] |
paul@95 | 104 | |
paul@95 | 105 | # Treat each part of the verified, possibly encrypted message as a file. |
paul@95 | 106 | |
paul@95 | 107 | for file in files: |
paul@95 | 108 | data = file.get_payload(decode=True) |
paul@95 | 109 | |
paul@104 | 110 | realname = file["Content-Disposition"] |
paul@104 | 111 | filename = join(directory, realname) |
paul@95 | 112 | |
paul@95 | 113 | dirpath = join(target_dir, directory) |
paul@95 | 114 | if not exists(dirpath): |
paul@95 | 115 | makedirs(dirpath) |
paul@95 | 116 | writefile(join(target_dir, filename), data) |
paul@95 | 117 | |
paul@95 | 118 | # vim: tabstop=4 expandtab shiftwidth=4 |