Mercurial > hg > bitsyblog

annotate bitsyblog/blogme.py @ 43:8b9aa9f40fa5

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
moving most of blogme to its own function
author k0s <k0scist@gmail.com>
date Wed, 11 Nov 2009 19:25:21 -0500
parents ffe95e34066f
children 0e7f56709c90
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
1 #!/usr/bin/env python
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
2
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
3 import optparse
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
4 import os
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
5 import subprocess
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
6 import sys
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
7 import tempfile
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
8 import urllib2
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
9
43
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
10 # global variable
0
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
11 dotfile='.blogme'
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
12
43
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
13 def tmpbuffer(editor=None):
0
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
14 """open an editor and retreive the resulting editted buffer"""
43
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
15 if not editor:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
16 editor = os.environ['EDITOR']
0
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
17 tmpfile = tempfile.mktemp(suffix='.txt')
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
18 cmdline = editor.split()
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
19 cmdline.append(tmpfile)
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
20 edit = subprocess.call(cmdline)
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
21 buffer = file(tmpfile).read().strip()
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
22 os.remove(tmpfile)
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
23 return buffer
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
24
43
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
25 def main(args=sys.argv[1:]):
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
26
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
27 # create the option parser
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
28 parser = optparse.OptionParser()
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
29 parser.add_option('-s', '--server')
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
30 parser.add_option('-u', '--user')
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
31 parser.add_option('-p', '--password')
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
32 parser.add_option('--private', action='store_true', default=False)
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
33 parser.add_option('--secret', action='store_true', default=False)
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
34
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
35 options, args = parser.parse_args()
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
36
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
37 # sanity check
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
38 if options.private and options.secret:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
39 print "post can't be secret and private!"
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
40 sys.exit(1)
0
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
41
43
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
42 # parse dotfile
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
43 home = os.environ.get('HOME')
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
44 if home:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
45 dotfile = os.path.join(home, dotfile)
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
46 if os.path.exists(dotfile):
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
47 prefs = file(dotfile).read().split('\n')
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
48 prefs = [ i for i in prefs if i.strip() ]
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
49 prefs = [ [ j.strip() for j in i.split(':', 1) ] for i in prefs
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
50 if ':' in i] # probably not necessary
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
51 prefs = dict(prefs)
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
52 else:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
53 prefs = {}
0
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
54
43
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
55 # determine user name and password
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
56 fields = [ 'user', 'password' ]
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
57 for field in fields:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
58 globals()[field] = prefs.get(field)
0
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
59
43
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
60 optval = getattr(options, field)
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
61 if optval:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
62 password = None # needed to ensure prompting for pw from command line
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
63 globals()[field] = optval
0
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
64
43
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
65 if globals()[field] is None:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
66 globals()[field] = raw_input('%s: ' % field)
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
67 assert user is not None
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
68 assert password is not None
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
69
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
70 # write the dotfile if it doesn't exist
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
71 if not os.path.exists(dotfile):
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
72 preffile = file(dotfile, 'w')
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
73 print >> preffile, 'user: %s' % user
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
74 print >> preffile, 'password: %s' % password
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
75 preffile.close()
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
76 os.chmod(dotfile, 0600)
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
77
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
78
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
79 # get the blog
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
80 if args:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
81 msg = ' '.join(args)
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
82 else:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
83 msg = tmpbuffer()
0
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
84
43
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
85 # open the url
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
86 url = '/'.join((options.server, user))
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
87 url += '?auth=digest' # specify authentication method
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
88 if options.private:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
89 url += '&privacy=private'
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
90 if options.secret:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
91 url += '&privacy=secret'
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
92 authhandler = urllib2.HTTPDigestAuthHandler()
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
93 authhandler.add_password('bitsyblog', url, user, password)
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
94 opener = urllib2.build_opener(authhandler)
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
95 urllib2.install_opener(opener)
0
e3823be6a423 initial commit of bitsyblog, from https://svn.openplans.org/svn/standalone/bitsyblog/trunk/
k0s <k0scist@gmail.com>
parents:
diff changeset
96
43
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
97 try:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
98 url = urllib2.urlopen(url, data=msg)
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
99 print url.url # print the blog post's url
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
100 except urllib2.HTTPError:
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
101 pass
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
102
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
103 if __name__ == '__main__':
8b9aa9f40fa5 moving most of blogme to its own function
k0s <k0scist@gmail.com>
parents: 42
diff changeset
104 main()