Initial commit

Signed-off-by: Johannes Löthberg <johannes@kyriasis.com>

1 files changed, 113 insertions, 0 deletions

diff --git a/fingerchange.py b/fingerchange.py
new file mode 100644
index 0000000..4217cd4
--- /dev/null
+++ b/fingerchange.py
@@ -0,0 +1,113 @@
+from textwrap import fill, indent
+from bs4 import BeautifulSoup
+import requests
+import asyncio
+import json
+
+api_url = 'http://api.stackexchange.com/2.2/users/'
+
+
+def get_user_by_uid(site, uid):
+    params = {
+        'site': site,
+        'filter': '!9YdnSA07B',
+    }
+    res = requests.get(api_url + str(uid), params=params)
+    res = json.loads(res.text)
+
+    return res['items'][0]
+
+
+def get_users(site, name=''):
+    params = {
+        'site': site,
+        'inname': name,
+        'filter': '!9YdnSA07B',
+    }
+    res = requests.get(api_url, params=params)
+    res = json.loads(res.text)
+
+    return res['items']
+
+
+def format_url(url):
+    return '<{}>'.format(url)
+
+
+def format_field(key, value):
+    return ' :: {} {}\n'.format(key, value)
+
+
+def format_header(header, filler):
+    header = str(header)
+    header_len = len(header)
+    filler_len = 80 - header_len - (3 + 2)
+
+    msg = '{pre_fil} {header} {post_fil}\n'.format(**{
+        'pre_fil': filler * 3,
+        'header': header,
+        'post_fil': filler * filler_len,
+    })
+    return msg
+
+
+def format_user(user):
+    msg = '\n'
+    msg += format_header(user['account_id'], '=')
+    msg += format_field('uid', user['user_id'])
+    msg += format_field('name', user['display_name'])
+    msg += format_field('reputation', user['reputation'])
+    msg += format_field('profile', format_url(user['link']))
+    if 'website_url' in user:
+        msg += format_field('website', format_url(user['website_url'])) + '\n'
+    else:
+        msg += '\n'
+
+    if 'about_me' in user:
+        msg += format_header('About User', '-') + '\n'
+
+        soup = BeautifulSoup(user['about_me'])
+        text = soup.get_text()
+
+        lines = [ line for line in text.splitlines() if line.strip() ]
+        for line in lines:
+            msg += indent(fill(line), '  ') + '\n\n'
+
+    return msg
+
+
+@asyncio.coroutine
+def handle_client(reader, writer):
+    line = yield from reader.readline()
+    line = line.decode('utf-8').strip()
+
+    if '@' not in line:
+        writer.write('Error: Username or site missing\n'.encode('utf-8'))
+        writer.close()
+        return
+
+    user, site = line.split('@')
+    try:
+        uid = int(user)
+        users = [get_user_by_uid(site, uid)]
+    except ValueError:
+        users = get_users(site, user)
+
+    for user in users:
+        msg = format_user(user)
+        writer.write(msg.encode('utf-8'))
+
+    writer.close()
+
+
+loop = asyncio.get_event_loop()
+
+coro = asyncio.start_server(handle_client, host='127.0.0.1', port=79, loop=loop)
+server = loop.run_until_complete(coro)
+
+try:
+    loop.run_forever()
+except KeyboardInterrupt:
+    server.close()
+    loop.run_until_complete(server.wait_closed())
+    loop.close()