Vitor Hideyoshi
4ae8385918
Adds the Environment, External, Utils folder inside de DPpack. All classes are going to be implemented there
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)) |