Skip to content

Instantly share code, notes, and snippets.

@kenjoe41

kenjoe41/gister.py

Created May 22, 2015 09:33
Show Gist options
  • Save kenjoe41/33c75e15a0f7c686de8e to your computer and use it in GitHub Desktop.
Save kenjoe41/33c75e15a0f7c686de8e to your computer and use it in GitHub Desktop.
Download ZIP
created by github.com/tr3buchet/gister
Raw
gister.py
#!/usr/bin/env python
#
# Copyright 2013 Trey Morris
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
import argparse
import ConfigParser
import json
import os
import re
import requests
import sys
# keyring is optional
try:
import keyring
except ImportError:
keyring = None
def parse_arguments():
parser = argparse.ArgumentParser(description='make gists!')
parser.add_argument('-p', '--private', action='store_true',
help='put gist on configured enterprise github')
parser.add_argument('-s', '--secret', action='store_true',
help='gist will be secret (not public)')
parser.add_argument('-a', '--anonymous', action='store_true',
help='gist will be anonymous')
parser.add_argument('-c', '--command', action='store',
help='command to prepend to gist')
parser.add_argument('-v', '--vim', action='store_true',
help='gist came from vim, no prompt/history')
parser.add_argument('file', nargs='*', action='store',
help='name of file(s) to gist')
parser.add_argument('-d', '--description', help='description of the file.')
return parser.parse_args()
def parse_config():
config_parser = ConfigParser.RawConfigParser(
defaults={'prompt': '[%(username)s|%(hostname)s %(cwd)s]%%',
'public_github_url': 'https://api.github.com',
'private_github_url': None,
'public_oauth': None,
'private_oauth': None})
# read from ~/.gister if it exists else use defaults
config_parser.read(os.path.expanduser('~/.gister'))
if config_parser.has_section('gister'):
items = config_parser.items('gister')
else:
items = config_parser.items('DEFAULT')
config = {}
for name, value in items:
# add in tokens from keyring if necessary
if 'oauth' in name and value == 'KEYRING':
if keyring:
value = keyring.get_password('gister', name)
else:
raise Exception('KEYRING specified in config but '
'python keyring package not available')
config[name] = value
return config
def get_stdin():
stdin_lines = []
for line in sys.stdin:
stdin_lines.append(line)
return ''.join(stdin_lines)
def get_filedata(filenames):
filedata = {}
for filename in filenames:
try:
with open(filename) as f:
filedata[filename.split('/')[-1]] = {'content': f.read()}
except IOError as e:
print e
sys.exit(1)
return filedata
def get_vim_payload():
lines = [line for line in sys.stdin]
filename = lines.pop(0).rstrip()
return {filename: {'content': ''.join(lines)}}
def get_commandline_payload(prompt=None, command=None, filenames=None):
if filenames:
return get_filedata(filenames)
if not (prompt and command):
return {'': {'content': get_stdin()}}
try:
username = os.environ['LOGNAME']
hostname = os.uname()[1]
cwd = re.sub('/home/%s' % username, '~', os.getcwd())
prompt = prompt % {'username': username,
'hostname': hostname, 'cwd': cwd}
except:
prompt = ''
if prompt:
prompt_command = '%s %s\n' % (prompt, command)
else:
prompt_command = ''
return {'': {'content': '%s%s' % (prompt_command, get_stdin())}}
def get_headers(conf, token_name):
return {'Authorization': 'token %s' % conf[token_name]}
def private_gist_url(conf, args):
if not conf['private_github_url']:
raise Exception('private_github_url must be set in ~/.gister '
'to create a gist on private github.\n'
'see https://github.com/tr3buchet/gister'
'#config-file---gister for more info on config '
'settings or create a gist on public github by '
'not specifying the -p flag')
if not args.anonymous and conf['private_oauth'] is None:
# non anonymous gists require oauth
raise Exception('private_oauth must be set in ~/.gister to create a gi'
'st on private github associated with your account.\n'
'see https://github.com/tr3buchet/gister'
'#config-file---gister for more info on config '
'settings or create an anonymous gist by '
'specifying the -a flag')
return conf['private_github_url']
def public_gist_url(conf, args):
if not args.anonymous and conf['public_oauth'] is None:
# non anonymous gists require oauth
raise Exception('public_oauth must be set in ~/.gister to create a '
'gist on public github associated with your account.\n'
'see https://github.com/tr3buchet/gister'
'#config-file---gister for more info on config '
'settings or create an anonymous gist by '
'specifying the -a flag')
return conf['public_github_url']
def create_gist():
args = parse_arguments()
conf = parse_config()
if args.vim:
payload = get_vim_payload()
else:
payload = get_commandline_payload(conf['prompt'], args.command,
args.file)
if args.private:
url = private_gist_url(conf, args)
token_name = 'private_oauth'
else:
url = public_gist_url(conf, args)
token_name = 'public_oauth'
description = 'created by github.com/tr3buchet/gister'
if args.description:
description = args.description
headers = get_headers(conf, token_name) if not args.anonymous else None
payload = {'description': description,
'public': not args.secret,
'files': payload}
# 'files': dict((k, {'content': v}) for k, v in payload)}
# 'files': {payload[0]: {'content': payload[1]}}}
try:
r = requests.post(url + '/gists', data=json.dumps(payload),
headers=headers)
except Exception as e:
print e
sys.exit()
r.raise_for_status()
print r.json()['html_url']
if __name__ == '__main__':
create_gist()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment