get-my-pastebin/get_my_pastebin.py
2019-01-14 17:08:30 +01:00

105 lines
3.5 KiB
Python

import click
from pick import pick
import urllib.request
import urllib.parse
import xml.etree.ElementTree as ET
import pyperclip
API_RESULTS_LIMIT = 1000
LOGIN_URL = 'https://pastebin.com/api/api_login.php'
API_LIST_URL = f'https://pastebin.com/api/api_post.php'
API_RAW_PASTE_URL = 'https://pastebin.com/api/api_raw.php'
@click.command()
@click.option('--api-key', required=True, help='API Dev key')
@click.option('--username', required=True, help='Your username account')
@click.option('--password', required=True, help='Your password account')
@click.argument('search')
def main(api_key, username, password, search):
""" Terminal application to find and copy your own Paste for Pastebin"""
user_key = get_user_key(api_key, username, password)
title = f'Results: {search}'
options = get_list(api_key, user_key, search)
if not options:
click.echo('Not results')
else:
# Show list to pick item
def get_label(option): return option.get('label')
option, index = pick(options, title, indicator='*', options_map_func=get_label)
raw_paste = get_raw_paste(api_key, user_key, option['paste_key'])
# Print raw paste
click.echo(raw_paste)
# Copy to clipboard
pyperclip.copy(raw_paste)
def get_user_key(api_key, username, password):
"""Get login key"""
post_url = 'http://localhost:3000/vim'
headers = {}
headers['Content-Type'] = 'application/json'
post_data = urllib.parse.urlencode(
{
'api_dev_key' : api_key,
'api_user_name': username,
'api_user_password': password
}
).encode('ascii')
post_response_user_key = urllib.request.urlopen(url=LOGIN_URL, data=post_data)
return post_response_user_key.read().decode("utf-8")
def get_list(api_key, user_key, search):
# Get all list
post_data = urllib.parse.urlencode(
{
'api_option': 'list',
'api_dev_key' : api_key,
'api_user_key': user_key,
'results_limit': API_RESULTS_LIMIT
}
).encode('ascii')
response_list = urllib.request.urlopen(url=API_LIST_URL, data=post_data)
data_list = '<list>' + response_list.read().decode("utf-8") + '</list>'
data_list = ET.fromstring(data_list)
# Search data
results = []
for paste in data_list:
found = False
label = ''
paste_key = ''
# Search in node. Save label and key
for item in paste:
if item.tag == 'paste_title' and search.upper() in item.text.upper():
found = True
label = item.text
if item.tag == 'paste_key':
paste_key = item.text
# Save node
if found:
results.append(
{
'label': label,
'paste_key': paste_key
}
)
return results
def get_raw_paste(api_key, user_key, paste_key):
# Get raw paste
post_data = urllib.parse.urlencode(
{
'api_option': 'show_paste',
'api_dev_key' : api_key,
'api_user_key': user_key,
'api_paste_key': paste_key
}
).encode('ascii')
response_paste = urllib.request.urlopen(url=API_RAW_PASTE_URL, data=post_data)
# Clear
raw_paste = response_paste.read().decode("utf-8").replace('\r', '')
return raw_paste
if __name__ == '__main__':
main()