def show_io(input_dir, output_dir):
swrite('\n=== DIRECTORIES ===\n\n')
# Show this directory
swrite("-- Current directory " + pwd() + ":\n")
write_list(ls('.'))
write_list(ls('./*'))
write_list(ls('./*/*'))
swrite("\n")
# List input and output directories
swrite("-- Input directory " + input_dir + ":\n")
write_list(ls(input_dir))
write_list(ls(input_dir + '/*'))
write_list(ls(input_dir + '/*/*'))
write_list(ls(input_dir + '/*/*/*'))
swrite("\n")
swrite("-- Output directory " + output_dir + ":\n")
write_list(ls(output_dir))
write_list(ls(output_dir + '/*'))
swrite("\n")
# write meta data to sdterr
swrite('\n=== METADATA ===\n\n')
swrite("-- Current directory " + pwd() + ":\n")
try:
metadata = yaml.load(open('metadata', 'r'))
for key,value in metadata.items():
swrite(key + ': ')
swrite(str(value) + '\n')
except:
swrite("none\n");
swrite("-- Input directory " + input_dir + ":\n")
try:
metadata = yaml.load(open(os.path.join(input_dir, 'metadata'), 'r'))
for key,value in metadata.items():
swrite(key + ': ')
swrite(str(value) + '\n')
swrite("\n")
except:
swrite("none\n");
python类write()的实例源码
def set_size(height, width):
stdout.write("\x1b[8;{rows};{cols}t".format(rows=height, cols=width))
return
def show_io(input_dir, output_dir):
''' show directory structure and inputs and autputs to scoring program'''
swrite('\n=== DIRECTORIES ===\n\n')
# Show this directory
swrite("-- Current directory " + pwd() + ":\n")
write_list(ls('.'))
write_list(ls('./*'))
write_list(ls('./*/*'))
swrite("\n")
# List input and output directories
swrite("-- Input directory " + input_dir + ":\n")
write_list(ls(input_dir))
write_list(ls(input_dir + '/*'))
write_list(ls(input_dir + '/*/*'))
write_list(ls(input_dir + '/*/*/*'))
swrite("\n")
swrite("-- Output directory " + output_dir + ":\n")
write_list(ls(output_dir))
write_list(ls(output_dir + '/*'))
swrite("\n")
# write meta data to sdterr
swrite('\n=== METADATA ===\n\n')
swrite("-- Current directory " + pwd() + ":\n")
try:
metadata = yaml.load(open('metadata', 'r'))
for key,value in metadata.items():
swrite(key + ': ')
swrite(str(value) + '\n')
except:
swrite("none\n");
swrite("-- Input directory " + input_dir + ":\n")
try:
metadata = yaml.load(open(os.path.join(input_dir, 'metadata'), 'r'))
for key,value in metadata.items():
swrite(key + ': ')
swrite(str(value) + '\n')
swrite("\n")
except:
swrite("none\n");
def write_scores(fp, scores):
''' Write scores to file opened under file pointer fp'''
for key in scores.keys():
fp.write("%s --> %s\n" % (key, scores[key]))
print(key + " --> " + str(scores[key]))
def write(filename, predictions):
''' Write prediction scores in prescribed format'''
with open(filename, "w") as output_file:
for row in predictions:
if type(row) is not np.ndarray and type(row) is not list:
row = [row]
for val in row:
output_file.write('{:g} '.format(float(val)))
output_file.write('\n')
def zipdir(archivename, basedir):
'''Zip directory, from J.F. Sebastian http://stackoverflow.com/'''
assert os.path.isdir(basedir)
with closing(ZipFile(archivename, "w", ZIP_DEFLATED)) as z:
for root, dirs, files in os.walk(basedir):
#NOTE: ignore empty directories
for fn in files:
if fn[-4:]!='.zip':
absfn = os.path.join(root, fn)
zfn = absfn[len(basedir)+len(os.sep):] #XXX: relative path
z.write(absfn, zfn)
# ================ Inventory input data and create data structure =================
def show_io(input_dir, output_dir):
swrite('\n=== DIRECTORIES ===\n\n')
# Show this directory
swrite("-- Current directory " + pwd() + ":\n")
write_list(ls('.'))
write_list(ls('./*'))
write_list(ls('./*/*'))
swrite("\n")
# List input and output directories
swrite("-- Input directory " + input_dir + ":\n")
write_list(ls(input_dir))
write_list(ls(input_dir + '/*'))
write_list(ls(input_dir + '/*/*'))
write_list(ls(input_dir + '/*/*/*'))
swrite("\n")
swrite("-- Output directory " + output_dir + ":\n")
write_list(ls(output_dir))
write_list(ls(output_dir + '/*'))
swrite("\n")
# write meta data to sdterr
swrite('\n=== METADATA ===\n\n')
swrite("-- Current directory " + pwd() + ":\n")
try:
metadata = yaml.load(open('metadata', 'r'))
for key,value in metadata.items():
swrite(key + ': ')
swrite(str(value) + '\n')
except:
swrite("none\n");
swrite("-- Input directory " + input_dir + ":\n")
try:
metadata = yaml.load(open(os.path.join(input_dir, 'metadata'), 'r'))
for key,value in metadata.items():
swrite(key + ': ')
swrite(str(value) + '\n')
swrite("\n")
except:
swrite("none\n");
def main():
"""Act like letsencrypt --version insofar as printing the version number to
stderr."""
if '--version' in argv:
stderr.write('letsencrypt 99.9.9\n')
def run(song, num, bearer=None, folder=None):
if song is None:
song = input(
"Enter the artist and song (e.g. The Beatles Let It Be): ")
youtubeLinks = []
if num == 1:
youtubeLinks.append(getYoutubeURLFromSearch(song))
elif bearer is None:
youtubeLinks = useLastFM(song, num)
else:
print("Using spotify")
youtubeLinks = useSpotify(song, num, bearer)
if len(youtubeLinks) == 0:
print("Could not find song recommendations for '%s'" % song)
return
# Start downloading and print out progress
if folder is not None:
chdir(folder)
else:
folder = abspath('.')
p = multiprocessing.Pool(multiprocessing.cpu_count())
print("\nStarting download...")
for i, _ in enumerate(p.imap_unordered(downloadURL, youtubeLinks), 1):
stderr.write(
'\r...{0:2.1%} complete'.format(i / len(youtubeLinks)))
print("\n\n%d tracks saved to %s\n" % (len(youtubeLinks), folder))
return
def latex2png(snippet, outfile):
"""Compiles a LaTeX snippet to png"""
pngimage = os.path.join(IMAGEDIR, outfile + '.png')
environment = os.environ
environment['openout_any'] = 'a'
environment['shell_escape_commands'] = \
"bibtex,bibtex8,kpsewhich,makeindex,mpost,repstopdf,gregorio"
proc = Popen(
["lualatex", '-output-directory=' + IMAGEDIR],
stdin=PIPE,
stdout=DEVNULL,
env=environment
)
proc.stdin.write(
(
LATEX_DOC % (snippet)
).encode("utf-8")
)
proc.communicate()
proc.stdin.close()
call(["pdfcrop", os.path.join(IMAGEDIR, "texput.pdf")], stdout=DEVNULL)
call(
[
"gs",
"-sDEVICE=pngalpha",
"-r144",
"-sOutputFile=" + pngimage,
os.path.join(IMAGEDIR, "texput-crop.pdf"),
],
stdout=DEVNULL,
)
def png(contents, latex_command):
"""Creates a png if needed."""
outfile = sha(contents + latex_command)
src = os.path.join(IMAGEDIR, outfile + '.png')
if not os.path.isfile(src):
try:
os.mkdir(IMAGEDIR)
stderr.write('Created directory ' + IMAGEDIR + '\n')
except OSError:
pass
latex2png(latex_command + "{" + contents + "}", outfile)
stderr.write('Created image ' + src + '\n')
return src
def ly2png(lily, outfile, staffsize):
p = Popen([
"lilypond",
"-dno-point-and-click",
"-dbackend=eps",
"-djob-count=2",
"-ddelete-intermediate-files",
"-o", outfile,
"-"
], stdin=PIPE, stdout=-3)
p.stdin.write(("\\paper{\n"
"indent=0\\mm\n"
"oddFooterMarkup=##f\n"
"oddHeaderMarkup=##f\n"
"bookTitleMarkup = ##f\n"
"scoreTitleMarkup = ##f\n"
"}\n"
"#(set-global-staff-size %s)\n" % staffsize +
lily).encode("utf-8"))
p.communicate()
p.stdin.close()
call([
"gs",
"-sDEVICE=pngalpha",
"-r144",
"-sOutputFile=" + outfile + '.png',
outfile + '.pdf',
], stdout=-3)
def png(contents, staffsize):
"""Creates a png if needed."""
outfile = os.path.join(IMAGEDIR, sha(contents + str(staffsize)))
src = outfile + '.png'
if not os.path.isfile(src):
try:
os.mkdir(IMAGEDIR)
stderr.write('Created directory ' + IMAGEDIR + '\n')
except OSError:
pass
ly2png(contents, outfile, staffsize)
stderr.write('Created image ' + src + '\n')
return src
def latex2png(snippet, outfile):
"""Compiles a LaTeX snippet to png"""
pngimage = os.path.join(IMAGEDIR, outfile + '.png')
environment = os.environ
environment['openout_any'] = 'a'
environment['shell_escape_commands'] = \
"bibtex,bibtex8,kpsewhich,makeindex,mpost,repstopdf,gregorio"
proc = Popen(
["lualatex", '-output-directory=' + IMAGEDIR],
stdin=PIPE,
stdout=DEVNULL,
env=environment
)
proc.stdin.write(
(
LATEX_DOC % (snippet)
).encode("utf-8")
)
proc.communicate()
proc.stdin.close()
call(["pdfcrop", os.path.join(IMAGEDIR, "texput.pdf")], stdout=DEVNULL)
call(
[
"gs",
"-sDEVICE=pngalpha",
"-r144",
"-sOutputFile=" + pngimage,
os.path.join(IMAGEDIR, "texput-crop.pdf"),
],
stdout=DEVNULL,
)
def ly2png(lily, outfile, staffsize):
p = Popen([
"lilypond",
"-dno-point-and-click",
"-dbackend=eps",
"-djob-count=2",
"-ddelete-intermediate-files",
"-o", outfile,
"-"
], stdin=PIPE, stdout=-3)
p.stdin.write(("\\paper{\n"
"indent=0\\mm\n"
"oddFooterMarkup=##f\n"
"oddHeaderMarkup=##f\n"
"bookTitleMarkup = ##f\n"
"scoreTitleMarkup = ##f\n"
"}\n"
"#(set-global-staff-size %s)\n" % staffsize +
lily).encode("utf-8"))
p.communicate()
p.stdin.close()
call([
"gs",
"-sDEVICE=pngalpha",
"-r144",
"-sOutputFile=" + outfile + '.png',
outfile + '.pdf',
], stdout=-3)
def png(contents, staffsize):
"""Creates a png if needed."""
outfile = os.path.join(IMAGEDIR, sha(contents + str(staffsize)))
src = outfile + '.png'
if not os.path.isfile(src):
try:
os.mkdir(IMAGEDIR)
stderr.write('Created directory ' + IMAGEDIR + '\n')
except OSError:
pass
ly2png(contents, outfile, staffsize)
stderr.write('Created image ' + src + '\n')
return src
def ask_password():
try:
raw_pass = os.environ['TEMBOARD_UI_PASSWORD']
except KeyError:
raw_pass = getpass(" Password: ")
try:
password = raw_pass
validate_parameters({'password': password},
[('password', T_PASSWORD, False)])
except HTTPError:
stdout.write("Invalid password.\n")
return ask_password()
return password
def ask_username():
try:
raw_username = os.environ['TEMBOARD_UI_USER']
except KeyError:
raw_username = raw_input(" Username: ")
try:
username = raw_username
validate_parameters({'username': username},
[('username', T_USERNAME, False)])
except HTTPError:
stdout.write("Invalid username.\n")
return ask_username()
return username
def main():
job_id = 0
f_log = open('./dummy_cuckoo.log', 'a')
while True:
timestamp = datetime.utcnow().strftime('%Y-%m-%d %H:%M:%S,%f')[:-3]
msg = '%s [lib.cuckoo.core.scheduler] INFO: Task #%d: reports generation completed ...\n' % (timestamp, job_id)
f_log.write(msg)
stderr.write(msg)
sleep(1)
# gcd(p, q) == 1
# m = (p * q) - 1
# n = (n + 5) % m
job_id = (job_id + 5) % 14
f_log.flush()
f_log.close()
def main():
"""Act like letsencrypt --version insofar as printing the version number to
stderr."""
if '--version' in argv:
stderr.write('letsencrypt 99.9.9\n')
