-
-
Save omz/b0644f5ed1d94bd32805 to your computer and use it in GitHub Desktop.
New from Gist
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# This script downloads and opens a Gist from a URL in the clipboard. | |
# It's meant to be put in the editor's actions menu. | |
# | |
# It works with "raw" and "web" gist URLs, but not with gists that | |
# contain multiple files or non-Python files. | |
# | |
# If a file already exists, a dialog is shown that asks whether the | |
# new file should be renamed automatically. | |
import clipboard | |
import editor | |
import console | |
import re | |
import os | |
class InvalidGistURLError (Exception): pass | |
class MultipleFilesInGistError (Exception): pass | |
class NoFilesInGistError (Exception): pass | |
class GistDownloadError (Exception): pass | |
def download_gist(gist_url): | |
# Returns a 2-tuple of filename and content | |
# console.show_activity() | |
raw_match = re.match('http(s?)://raw.github.com/gist/', gist_url) | |
if raw_match: | |
import requests | |
from urlparse import urlparse | |
filename = os.path.split(urlparse(gist_url).path)[1] | |
try: | |
r = requests.get(gist_url) | |
content = r.text | |
return filename, content | |
except: | |
raise GistDownloadError() | |
else: | |
gist_id_match = re.match('http(s?)://gist.github.com/([0-9a-zA-Z]*)/([0-9a-f]*)', gist_url) | |
if gist_id_match: | |
import requests | |
gist_id = gist_id_match.group(3) | |
json_url = 'https://api.github.com/gists/' + gist_id | |
try: | |
import json | |
gist_json = requests.get(json_url).text | |
gist_info = json.loads(gist_json) | |
files = gist_info['files'] | |
except: | |
raise GistDownloadError() | |
py_files = [] | |
for file_info in files.values(): | |
lang = file_info.get('language', None) | |
if lang != 'Python': | |
continue | |
py_files.append(file_info) | |
if len(py_files) > 1: | |
raise MultipleFilesInGistError() | |
elif len(py_files) == 0: | |
raise NoFilesInGistError() | |
else: | |
file_info = py_files[0] | |
filename = file_info['filename'] | |
content = file_info['content'] | |
return filename, content | |
else: | |
raise InvalidGistURLError() | |
def main(): | |
gist_url = clipboard.get() | |
try: | |
filename, content = download_gist(gist_url) | |
if os.path.isfile(filename): | |
i = console.alert('File exists', 'A file with the name ' + filename + | |
' already exists in your library.', | |
'Auto Rename') | |
if i == 1: | |
editor.make_new_file(filename, content) | |
else: | |
editor.make_new_file(filename, content) | |
except InvalidGistURLError: | |
console.alert('No Gist URL', | |
'The clipboard doesn\'t seem to contain a valid Gist URL.', | |
'OK') | |
except MultipleFilesInGistError: | |
console.alert('Multiple Files', 'This Gist contains multiple ' + | |
'Python files, which isn\'t currently supported.') | |
except NoFilesInGistError: | |
console.alert('No Python Files', 'This Gist contains no Python files.') | |
except GistDownloadError: | |
console.alert('Error', 'The Gist could not be downloaded.') | |
if __name__ == '__main__': | |
main() |
Why not support multiple files in gist? That might be very helpful when downloading a lib of my own, for instance
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I needed to change row 41 to gist_id_match = re.match('http(s?)://gist.github.com/([0-9a-zA-Z])/([0-9a-f])', gist_url)
and row 44 to be gist_id = gist_id_match.group(3)