async def endpoint(site: str):
"""__[Web-SS Endpoint]__
__Args:__
`site (str)`: `[website url]`
__Returns:__
`[type]`: `[FileResponse]` = image: mime_type: image/png
__Exception:__
`[error]`: `[pyppeteer.errors]`
"""
global browser
if not is_browser_started:
print("started")
await start_browser()
is_browser_started = True
page = await browser.newPage()
file = tempfile() + ".png"
await page.setViewport({'width': 2000, 'height': 1381, 'deviceScaleFactor': 2.0})
try:
await page.goto(site)
except pyppeteer.errors.NetworkError as e:
return {
"error": str(e)
}
await page.screenshot({'path': file})
await page.close()
return FileResponse(file)
def run(applescript, background=False):
path = applescript
if not os.path.exists(applescript): # source code
path = temp.tempfile()
open(path, "w").write(applescript)
args = ["osascript", path]
return runcmd.run(args, background=background)
def policy(self):
path = temp.tempfile()
open(path, "w").write(POLICY % settings.AWS_STATIC_BUCKET)
args = [
"s3api", "put-bucket-policy", "--bucket",
settings.AWS_STATIC_BUCKET, "--policy",
"file://%s" % path
]
aws_cli(args)
def qpixmap_to_cvmat(img: QPixmap) -> np.array:
try:
path = temp.tempfile() + '.png'
img.save(path)
mat = cv2.imread(path, cv2.IMREAD_COLOR)
finally:
if os.path.exists(path):
os.remove(path)
return mat
def subopen(args, stdin=None, shell=False, cwd=None, env=None):
temp = None
pid = None
close_fds = True
rstrip = True
try:
if stdin is not None and isstring(stdin):
temp = tempfile()
write(temp, stdin)
stdin = open(temp)
args = tolist(args)
for i, arg in enumerate(args):
if not isinstance(arg, int):
if PY3:
args[i] = str(arg) # python3
else:
args[i] = __builtin__.unicode(arg) # python2
if not env:
env = os.environ.copy()
process = subprocess.Popen(args,
stdin=stdin,
stderr=subprocess.PIPE,
stdout=subprocess.PIPE,
close_fds=close_fds,
shell=shell,
cwd=cwd,
env=env,
universal_newlines=False # ,
# startupinfo=startupinfo#,
# creationflags=creationflags
)
pid = process.pid
stdoutdata, stderrdata = process.communicate()
if PY3:
# python3. process.communicate returns bytes
stdoutdata = str(stdoutdata, "utf-8") # stdoutdata
stderrdata = str(stderrdata, "utf-8") # stderrdata
# rstrip
if rstrip and stdoutdata:
stdoutdata = stdoutdata.rstrip()
if rstrip and stderrdata:
stderrdata = stderrdata.rstrip()
returncode = process.returncode
return (returncode, stdoutdata, stderrdata)
# raise CalledProcessError(returncode,args,err)
# if returncode!=0:
# raise CalledProcessError(returncode,args,stderrdata)
# return stdoutdata
finally:
try:
if pid:
os.killpg(pid, signal.SIGTERM)
except Exception:
pass
# print(format_exc())
rm(temp)
def run(applescript, background=False):
"""run applescript file/code"""
if os.path.exists(applescript):
path = applescript
else:
path = temp.tempfile()
open(path, "w").write(applescript)
cmd = ["osascript", path]
r = runcmd.run(cmd, background=background)
return r.code, r.out, r.err
def cvmat_to_qpixmap(mat: np.array) -> QPixmap:
try:
tmpfn = temp.tempfile() + '.png'
cv2.imwrite(tmpfn, mat)
img = QPixmap()
img.load(tmpfn)
finally:
if os.path.exists(tmpfn):
os.remove(tmpfn)
return img
def loadstl(self, stldata):
try:
self.stldata = stldata
tmpfn = temp.tempfile() + '.stl'
with open(tmpfn, "wb") as fh:
fh.write(stldata)
vtkw = self._setupVtk(tmpfn)
self.addWidget(vtkw)
self.movie.stop()
self.setCurrentWidget(vtkw)
finally:
if os.path.exists(tmpfn):
os.remove(tmpfn)
def loadstl(self, stldata):
try:
tmpfn = temp.tempfile() + '.stl'
with open(tmpfn, "wb") as fh:
fh.write(stldata)
self.reader.SetFileName(tmpfn)
self.mapper.SetInputConnection(self.reader.GetOutputPort())
self.reader.Update()
self.ren.ResetCamera()
self.update()
finally:
if os.path.exists(tmpfn):
os.remove(tmpfn)
def test_file_update(self):
v = Version(0, 4, 2, "")
temp_filename = temp.tempfile()
try:
v.write(temp_filename)
v2 = Version(0, 4, 3, "")
Version.update(temp_filename, v2)
v3 = Version().read(temp_filename)
self.assertEqual(v2, v3)
self.assertNotEqual(v, v3)
finally:
if os.path.isfile(temp_filename):
os.unlink(temp_filename)
def generate_firmware_for_setup_drone(self):
try:
code_gen_file = open("./input_code/Main.ino", "r", encoding="UTF-8")
raw_data = code_gen_file.readlines()
settings = []
if (self.checkBox.isChecked()): # debug
settings.append("#define DEBUG 1 \n")
else:
settings.append("#define DEBUG 0 \n")
if (self.checkBox_2.isChecked()):
settings.append("#define MANUAL_SUPPORT 1 \n")
else:
settings.append("#define MANUAL_SUPPORT 0\n")
if (self.checkBox_3.isChecked()):
settings.append("#define MANUAL_API_LOGO 1\n")
else:
settings.append("#define MANUAL_API_LOGO 0\n")
if (self.checkBox_4.isChecked()):
settings.append("#define ULTROSONICS_SUPPORT 1\n")
else:
settings.append("#define ULTROSONICS_SUPPORT 0\n")
if (self.checkBox_5.isChecked()):
settings.append("#define BUZZER_SUPPORT 1\n")
else:
settings.append("#define BUZZER_SUPPORT 0\n")
if (self.checkBox_6.isChecked()):
settings.append("boolean auto_level = true; //Auto level on (true) or off (false) \n")
else:
settings.append("boolean auto_level = false; //Auto level on (true) or off (false) \n")
final_massive = settings + raw_data
out_path=temp.tempfile()
file = open(out_path, "a", encoding="UTF-8")
for data_out in range(len(final_massive)):
file.write(final_massive[data_out])
file.flush()
file.close()
file = open(out_path, 'r')
with file:
text = file.read()
self.plainTextEdit_3.appendPlainText(text)
except Exception as error:
print(error)
async def endpoint(site: str):
"""__[Web-SS Endpoint]__
__Args:__
`site (str)`: `[website url]`
__Returns:__
`[type]`: `[FileResponse]` = image: mime_type: image/png
__Exception:__
`[error]`: `[pyppeteer.errors]`
"""
browser = await pyppeteer.launch(headless=True)
page = await browser.newPage()
file = tempfile() + ".png"
try:
await page.goto(site)
except pyppeteer.errors.NetworkError as e:
return {"error": str(e)}
await page.screenshot({'path': file})
return FileResponse(file)
def osascript(applescript, flags=None):
"""osascript applescript code or file
"""
args = ["osascript"]
if applescript and isinstance(applescript, list):
applescript = "\n".join(applescript)
temp = programfile = None
if applescript and os.path.exists(applescript):
programfile = applescript
else:
temp = programfile = tempfile()
write(temp, applescript)
try:
if flags:
args += ["-s", flags]
if programfile:
args.append(programfile)
returncode, stdout, stderr = subopen(args)
return (returncode, stdout, stderr)
finally:
rm(temp)
def test_file_write(self):
temp_filename = temp.tempfile()
try:
v = Version(0, 4, 1, "beta")
v.write(temp_filename)
x = Version().read(temp_filename)
self.assertEqual(x, v)
x.bump_tag()
x.write(temp_filename)
w = Version().read(temp_filename)
self.assertEqual(x, w)
v = Version(0, 3, 1, 'beta') # test single quotes for tag
v.write(temp_filename)
x = Version().read(temp_filename)
self.assertEqual(x, v)
finally:
if os.path.isfile(temp_filename):
os.unlink(temp_filename)
def _tempfile(content):
path = temp.tempfile()
open(path, "w").write(content)
return path