rplaceimage.py

import math
import sys
import time
import random
import struct

import requests
from PIL import Image
from requests.adapters import HTTPAdapter

img = Image.open(sys.argv[1])
origin = (int(sys.argv[2]), int(sys.argv[3]))
username = sys.argv[4]
password = sys.argv[5]

def find_palette(point):
    rgb_code_dictionary = {
        (255, 255, 255): 0,
        (228, 228, 228): 1,
        (136, 136, 136): 2,
        (34, 34, 34): 3,
        (255, 167, 209): 4,
        (229, 0, 0): 5,
        (229, 149, 0): 6,
        (160, 106, 66): 7,
        (229, 217, 0): 8,
        (148, 224, 68): 9,
        (2, 190, 1): 10,
        (0, 211, 211): 11,
        (0, 131, 199): 12,
        (0, 0, 234): 13,
        (207, 110, 228): 14,
        (130, 0, 128): 15
    }

    def distance(c1, c2):
        (r1, g1, b1) = c1
        (r2, g2, b2) = c2
        return math.sqrt((r1 - r2) ** 2 + (g1 - g2) ** 2 + (b1 - b2) ** 2)

    colors = list(rgb_code_dictionary.keys())
    closest_colors = sorted(colors, key=lambda color: distance(color, point))
    closest_color = closest_colors[0]
    code = rgb_code_dictionary[closest_color]
    return code

s = requests.Session()
s.mount('https://www.reddit.com', HTTPAdapter(max_retries=5))
s.headers["User-Agent"] = "PlacePlacer"
r = s.post("https://www.reddit.com/api/login/{}".format(username),
           data={"user": username, "passwd": password, "api_type": "json"})
s.headers['x-modhash'] = r.json()["json"]["data"]["modhash"]

bitmp = []

for ay in range(img.height):
    bitmp.append([])
    for ax in range(img.width):
        bitmp[ay].append(find_palette(img.getpixel((ax,ay))))

print "Data preperation done, pushing pixels."

def get_bad_pixels():
    badpixels = []
    r = s.get("https://www.reddit.com/api/place/board-bitmap", timeout=10)
    boardcols = r.content[4:]
    ay = 0
    for ay in range(len(bitmp)):
        for ax in range(len(bitmp[ay])):
            boardpos = (origin[1] + ay) * 1000 + origin[0] + ax
            boardcol = struct.unpack('B', boardcols[int(boardpos/2)])[0]
            if boardpos % 2 == 0:
                boardcol = boardcol >> 4
            else:
                boardcol = boardcol & 15
            if int(bitmp[ay][ax]) != boardcol:
                badpixels.append((ax,ay))
    return badpixels

while True:
    badpixels = get_bad_pixels()
    while len(badpixels) == 0:
        time.sleep(60)
        badpixels = get_bad_pixels()
    badpixel = random.choice(badpixels)
    ax = badpixel[0]
    ay = badpixel[1]
    print("Placing color {} at {},{}".format(bitmp[ay][ax], origin[0] + ax, origin[1] + ay))
    r = s.post("https://www.reddit.com/api/place/draw.json",
            data={"x": str(origin[0] + ax), "y": str(origin[1] + ay), "color": bitmp[ay][ax]})
    try:
        secs = float(r.json()["wait_seconds"])
    except:
        print("Exception in script - {}, retrying in 10 secs".format(r.json()))
        secs = 10
    waitTime = int(secs) + 2
    if "error" not in r.json():
        print("Placed color, waiting {} seconds.".format(waitTime))
    else:
        print("Cooldown already active, waiting {} seconds.".format(waitTime))
    while(waitTime > 0):
        time.sleep(1)
        waitTime -= 1