105 lines
3.5 KiB
Python
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()
|