mirror of
https://github.com/captbaritone/webamp.git
synced 2026-07-22 01:29:22 +00:00
150 lines
4.2 KiB
Python
Executable file
150 lines
4.2 KiB
Python
Executable file
#!/usr/bin/env python
|
|
"""Tweet Winamp Skins
|
|
|
|
Usage:
|
|
tweet.py [--dry]
|
|
|
|
Options:
|
|
-h --help Show this screen.
|
|
--version Show version.
|
|
--dry Don't actually post a tweet
|
|
"""
|
|
|
|
import requests
|
|
import os
|
|
import random
|
|
import json
|
|
import urllib
|
|
import hashlib
|
|
import boto3
|
|
import twitter
|
|
from docopt import docopt
|
|
from sets import Set
|
|
|
|
from config import CONFIG
|
|
|
|
|
|
def tweet(text, img_path):
|
|
api = twitter.Api(
|
|
consumer_key=CONFIG['consumer_key'],
|
|
consumer_secret=CONFIG['consumer_secret'],
|
|
access_token_key=CONFIG['access_token_key'],
|
|
access_token_secret=CONFIG['access_token_secret'])
|
|
api.PostMedia(text, img_path)
|
|
|
|
|
|
def find(dir):
|
|
for root, sub_folders, files in os.walk(dir):
|
|
for file in files:
|
|
yield os.path.join(root, file)
|
|
|
|
|
|
def random_skin():
|
|
return random.choice([f for f in find("../automatedScreenshots") if f.endswith(".wsz")])
|
|
|
|
|
|
def screenshot_by_md5(md5):
|
|
screenshots = find("../automatedScreenshots/screenshots/")
|
|
matchin_screenshots = [
|
|
f for f in screenshots if f.endswith("%s.png" % md5)]
|
|
if(len(matchin_screenshots) is not 1):
|
|
return None
|
|
return matchin_screenshots[0]
|
|
|
|
|
|
def url_is_good(url):
|
|
try:
|
|
r = requests.head(url)
|
|
return r.status_code == 200
|
|
# prints the int of the status code. Find more at httpstatusrappers.com :)
|
|
except Exception:
|
|
return False
|
|
|
|
|
|
def md5_file(path):
|
|
hash_md5 = hashlib.md5()
|
|
with open(path, "rb") as f:
|
|
for chunk in iter(lambda: f.read(4096), b""):
|
|
hash_md5.update(chunk)
|
|
return hash_md5.hexdigest()
|
|
|
|
|
|
def find_skin_with_screenshot():
|
|
skin_path = random_skin()
|
|
|
|
md5 = md5_file(skin_path)
|
|
screenshot_path = screenshot_by_md5(md5)
|
|
if(not screenshot_path):
|
|
print "Could not find the screenshot, trying again..."
|
|
return find_skin_with_screenshot()
|
|
return skin_path, screenshot_path, md5
|
|
|
|
|
|
def dispatch(action):
|
|
with open("./action_log.json", 'r') as f:
|
|
action_log = json.load(f)
|
|
action_log.append(action)
|
|
with open("./action_log.json", 'w') as f:
|
|
json.dump(action_log, f)
|
|
|
|
|
|
def past_skins():
|
|
with open("./action_log.json", 'r') as f:
|
|
action_log = json.load(f)
|
|
past = Set()
|
|
for action in action_log:
|
|
if(action['type'] == "REJECTED_SKIN" or action['type'] == "TWEETED"):
|
|
past.add(str(action["md5"]))
|
|
return past
|
|
|
|
|
|
def main(dry):
|
|
already_handled = past_skins()
|
|
skin_path, screenshot_path, md5 = find_skin_with_screenshot()
|
|
dispatch({"type": "FOUND_SCREENSHOT", "skin_path": skin_path,
|
|
"screenshot_path": screenshot_path, "md5": md5})
|
|
if(md5 in already_handled):
|
|
print "Already handled %s. Trying again..." % md5
|
|
return main(dry)
|
|
skin_name = os.path.basename(skin_path)
|
|
print "Found %s" % skin_name
|
|
os.system("open \"%s\"" % screenshot_path)
|
|
res = raw_input("continue?")
|
|
if(res is not "y"):
|
|
dispatch({"type": "REJECTED_SKIN", "md5": md5})
|
|
return main(dry)
|
|
print "Screenshot: %s" % screenshot_path
|
|
|
|
print "Uploading to S3..."
|
|
s3 = boto3.resource('s3')
|
|
s3.meta.client.upload_file(
|
|
skin_path, 'winamp2-js-skins', skin_name, {'ACL': 'public-read'})
|
|
|
|
skin_url = "https://s3-us-west-2.amazonaws.com/winamp2-js-skins/%s" % skin_name
|
|
dispatch({"type": "UPLOADED_SKIN", "md5": md5, "skin_url": skin_url})
|
|
print "Done: %s" % skin_url
|
|
|
|
print "Going to check that URL..."
|
|
if not url_is_good(skin_url):
|
|
dispatch({"type": "FOUND_INVALID_URL",
|
|
"md5": md5, "skin_url": skin_url})
|
|
print "URL is no good. Aborting."
|
|
return
|
|
|
|
options = {"skinUrl": skin_url}
|
|
|
|
options_query = urllib.quote(json.dumps(options))
|
|
|
|
winamp2_js_url = "https://jordaneldredge.com/projects/winamp2-js/#%s" % options_query
|
|
|
|
status_message = """%s
|
|
Try: %s
|
|
Download: %s""" % (skin_name, winamp2_js_url, skin_url)
|
|
tweet(status_message, screenshot_path)
|
|
dispatch({"type": "TWEETED", "md5": md5, "message": status_message})
|
|
print "Done!"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
arguments = docopt(__doc__, version='Tweet Winamp Skins 0.1')
|
|
main(dry=arguments.get("--dry"))
|