Vitor Hideyoshi
2a4e9eff0c
This commit adds the methods that were present in the Gaussian.py file into the SetGlobals.py file and packages the program into a diceplayer module so it can be ran using 'python3 -m diceplayer' Signed-off-by: Vitor Hideyoshi <vitor.h.n.batista@gmail.com>
67 lines
1.6 KiB
Python
67 lines
1.6 KiB
Python
import os, sys, time
|
|
from posixpath import sep
|
|
import shutil, gzip
|
|
|
|
####################################### functions ######################################
|
|
|
|
def weekday_date_time():
|
|
|
|
return time.strftime("%A, %d %b %Y at %H:%M:%S")
|
|
|
|
|
|
def date_time():
|
|
|
|
return time.strftime("%d %b %Y at %H:%M:%S")
|
|
|
|
|
|
def compress_files_1mb(path):
|
|
|
|
working_dir = os.getcwd()
|
|
os.chdir(path)
|
|
|
|
files = filter(os.path.isfile, os.listdir(os.curdir))
|
|
for file in files:
|
|
if os.path.getsize(file) > 1024 * 1024: ## If bigger than 1MB
|
|
filegz = file + ".gz"
|
|
try:
|
|
with open(file, 'rb') as f_in:
|
|
with gzip.open(filegz, 'wb') as f_out:
|
|
shutil.copyfileobj(f_in, f_out)
|
|
except:
|
|
sys.exit("Error: cannot compress file {}".format(file))
|
|
|
|
os.chdir(working_dir)
|
|
|
|
return
|
|
|
|
def make_simulation_dir():
|
|
|
|
sim_dir = "simfiles"
|
|
if os.path.exists(sim_dir):
|
|
sys.exit("Error: a file or a directory {} already exists, move or delete de simfiles directory to continue.".format(sim_dir))
|
|
try:
|
|
os.makedirs(sim_dir)
|
|
except:
|
|
sys.exit("Error: cannot make directory {}".format(sim_dir))
|
|
|
|
def make_step_dir(cycle):
|
|
|
|
sim_dir = "simfiles"
|
|
step_dir = "step{:02d}".format(cycle)
|
|
path = sim_dir + os.sep + step_dir
|
|
if os.path.exists(path):
|
|
sys.exit("Error: a file or directory {} already exists".format(step_dir))
|
|
try:
|
|
os.makedirs(path)
|
|
except:
|
|
sys.exit("Error: cannot make directory {}".format(step_dir))
|
|
|
|
def make_qm_dir(cycle):
|
|
|
|
sim_dir = "simfiles"
|
|
step_dir = "step{:02d}".format(cycle)
|
|
path = sim_dir + os.sep + step_dir + os.sep + "qm"
|
|
try:
|
|
os.makedirs(path)
|
|
except:
|
|
sys.exit("Error: cannot make directory {}".format(path)) |