1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
#!/usr/bin/python2.7
import os
import sys
import random
import re
import pb.lib.utils as utils
import urllib
import urlparse
from pb.config import *
import base64
import time
import string
import urllib
from subprocess import Popen, PIPE
import sha
import simplejson as json
import mimetypes
class Imlandscape(object):
def __init__(self, **kwargs):
self.params = {}
self.tag = "imlandscape"
self.commands = [];
self._required_keys = [
"heightmap",
"imgdata",
"texture",
"name",
]
self.now = utils.now()
self.files_created = []
for k in self._required_keys:
if k in kwargs:
self.params[k] = kwargs[k]
else:
self.params[k] = utils.bool_correct(utils.sanitize(kwargs[k]))
def _filename_from_url (self, url, name=""):
if "?" in url:
url = url.split("?")[0]
if "/" in url:
url = urllib.unquote(url).replace(" ","")
filename = url.split("/")[-1]
filetype = "png"
filename = sanitize(filename[:-4])
else:
filename = ""
if name != "":
name = name+"_"
return "{}_{}{}_{}.{}".format("imlandscape", name, filename,self.now, "png")
def _saveImgData(self, imgdata, filename):
try:
# up = urlparse.urlparse(url)
# head, data = imgdata.split(',', 1)
# bits = head.split(';')
parts = imgdata.split(';')
mime_type = parts[0] if parts[0] else 'text/plain'
data = parts[1]
charset, b64 = 'ASCII', False
for part in parts[1]:
if part.startswith('charset='):
charset = part[8:]
elif part == 'base64':
b64 = True
# Do something smart with charset and b64 instead of assuming
if b64:
plaindata = base64.b64decode(data)
else:
plaindata = data
with open(filename, 'wb') as f:
f.write(plaindata)
except Exception as e:
sys.stderr.write("ERROR: {}\n".format(str(e)));
# def _cleanup(self):
# cmd = ["rm"]+self.files_created
# self._call_cmd(cmd)
def create(self, breakmode=""):
self.filepath = self._filename_from_url(self.params.get('texture',""), self.params.get('name',""));
self._saveImgData(self.params.get('imgdata'), self.filepath);
if __name__ == "__main__":
f = open("/tmp/base64img")
data = f.read()
f.close()
TEST_PARAMS = {
"heightmap" : "https%3A%2F%2Fwww.google.com%2Fimages%2Fsrpr%2Flogo11w.png" ,
"imgdata" : data,
"texture" : "https%3A%2F%2Fwww.google.com%2Fimages%2Fsrpr%2Flogo11w.png",
"name" : "pepper",
}
b = Imlandscape(**TEST_PARAMS)
b.create();
print b.filepath
|
