def test_create_file(s3_client):
# Create test.txt file
create_file("tests/test.txt", "hello world")
# get the object contents
object = s3_client.get_object(Bucket="bucket.proxyroot.com",
Key="tests/test.txt")
data = object["Body"].read().decode("utf-8")
assert data == "hello world"
_dpms,
'EndSection\n\n',
'Section "Device"\n',
' Identifier "Videocard0"\n',
' VendorName "Videocard vendor"\n',
' BoardName "%s"\n' % card,
' Driver "%s"\n' % driver,
_videoram,
_options,
'EndSection\n\n',
'Section "Screen"\n',
' Identifier "Screen0"\n',
' Device "Videocard0"\n',
' Monitor "Monitor0"\n',
' DefaultDepth %s\n' % depth,
' SubSection "Display"\n',
' Viewport 0 0\n',
' Depth %s\n' % depth,
' Modes "%s"\n' % resolution,
' EndSubSection\n',
'EndSection\n\n' ]
if (source.isRHEL() and source.cmpVersion("4") < 0) or \
(source.isFedora() and source.cmpVersion("2.9") < 0):
content.extend( ['Section "DRI"\n',
' Group 0\n',
' Mode 0666\n',
'EndSection\n' ] )
create_file(buildroot, conf, content)
# vim:ts=4:sw=4:showmatch:expandtab
def write_file(n): # not thread safe
name = junk(n)
now = datetime.datetime.utcnow().timestamp()
f.create_file(name, str([now]))
root = f.get_root()
path = f.get_path()
(filename, image_filename, json_filename,
label_filename) = f.get_my_filenames()
image_filename = filename
actual = []
if os.path.exists(image_filename):
print(image_filename + " exists")
print("reading:", image_filename)
actual = f.read_filenames_from_disk(image_filename)
else:
print(image_filename + " does not exist.")
print("examining camera folder for files")
good = get_filenames_from_camera_folders(path, root=root)
print(type(good), len(good), "file names")
print(good[0])
images = good
jsons = []
labels = []
if not (len(images) == len(jsons) and len(images) == len(labels)):
print("lengths of jsons and/or labels are not equal!")
#exit()
f.create_file(filename, good)
print("reading:", filename)
actual = f.read_filenames_from_disk(filename)
print("check")
if not f.is_ok(actual, good):
print("badness")
print(len(actual), "image file names")
# make into a function
def firewall_config(ks, buildroot, source):
if (source.isRHEL() and source.cmpVersion("4") < 0) or \
(source.isFedora() and source.cmpVersion("2") < 0) or \
not os.path.exists(buildroot+"/usr/sbin/lokkit"):
# lokkit is not able to configure firewall for pre RHEL-4 and
# pre FC-2
_trusted = ""
if ks["firewall"].has_key("trusted"):
for iface in ks["firewall"]["trusted"]:
_trusted += '-A RH-Firewall-1-INPUT -i %s -j ACCEPT\n' % \
iface
_open_ports = ""
if ks["firewall"].has_key("ports"):
ports = ks["firewall"]["ports"][:]
ports.sort()
for port in ports:
_open_ports += '-A RH-Firewall-1-INPUT ' + \
'-m state --state NEW ' + \
'-m %s -p %s --dport %d -j ACCEPT\n' % \
(port[1], port[1], port[0])
content = [ \
'# Firewall configuration written by pyrpmkickstart\n',
'# Manual customization of this file is not recommended.\n',
'*filter\n',
':INPUT ACCEPT [0:0]\n',
':FORWARD ACCEPT [0:0]\n',
':OUTPUT ACCEPT [0:0]\n',
':RH-Firewall-1-INPUT - [0:0]\n',
'-A INPUT -j RH-Firewall-1-INPUT\n',
'-A FORWARD -j RH-Firewall-1-INPUT\n',
'-A RH-Firewall-1-INPUT -i lo -j ACCEPT\n',
_trusted,
'-A RH-Firewall-1-INPUT -p icmp --icmp-type any -j ACCEPT\n',
'-A RH-Firewall-1-INPUT -p 50 -j ACCEPT\n',
'-A RH-Firewall-1-INPUT -p 51 -j ACCEPT\n',
'-A RH-Firewall-1-INPUT -p udp --dport 5353 -d 224.0.0.251 -j ACCEPT\n',
'-A RH-Firewall-1-INPUT -p udp -m udp --dport 631 -j ACCEPT\n',
'-A RH-Firewall-1-INPUT -m state --state ESTABLISHED,RELATED -j ACCEPT\n',
_open_ports,
'-A RH-Firewall-1-INPUT -j REJECT --reject-with icmp-host-prohibited\n',
'COMMIT\n' ]
create_file(buildroot, "/etc/sysconfig/iptables", content)
# enable firewall
if ks["firewall"].has_key("enabled"):
if run_script("/sbin/chkconfig iptables on", buildroot) != 0:
log.error("Could not enable firewall.")
else:
# use lokkit to configure firewall
fwargs = []
if ks["firewall"].has_key("enabled"):
fwargs.append("--enabled")
if ks["firewall"].has_key("disabled"):
fwargs.append("--disabled")
if ks["firewall"].has_key("trusted"):
for trusted in ks["firewall"]["trusted"]:
fwargs.append("--trusted=%s" % trusted)
if ks["firewall"].has_key("ports"):
for port in ks["firewall"]["ports"]:
fwargs.append("--port=%s:%s" % (port[0], port[1]))
lokkit = "/usr/sbin/lokkit --quiet --nostart -f %s" % \
" ".join(fwargs)
if run_script(lokkit, buildroot) != 0:
log.error("Configuration of firewall failed.")
create_file(buildroot, "/etc/sysconfig/system-config-securitylevel", [
'# Configuration file for system-config-securitylevel\n', "\n",
"%s" % "\n".join(fwargs)
])
def firewall_config(ks, buildroot, source):
if (source.isRHEL() and source.cmpVersion("4") < 0) or \
(source.isFedora() and source.cmpVersion("2") < 0) or \
not os.path.exists(buildroot+"/usr/sbin/lokkit"):
# lokkit is not able to configure firewall for pre RHEL-4 and
# pre FC-2
_trusted = ""
if ks["firewall"].has_key("trusted"):
for iface in ks["firewall"]["trusted"]:
_trusted += '-A RH-Firewall-1-INPUT -i %s -j ACCEPT\n' % \
iface
_open_ports = ""
if ks["firewall"].has_key("ports"):
ports = ks["firewall"]["ports"][:]
ports.sort()
for port in ports:
_open_ports += '-A RH-Firewall-1-INPUT ' + \
'-m state --state NEW ' + \
'-m %s -p %s --dport %d -j ACCEPT\n' % \
(port[1], port[1], port[0])
content = [ \
'# Firewall configuration written by pyrpmkickstart\n',
'# Manual customization of this file is not recommended.\n',
'*filter\n',
':INPUT ACCEPT [0:0]\n',
':FORWARD ACCEPT [0:0]\n',
':OUTPUT ACCEPT [0:0]\n',
':RH-Firewall-1-INPUT - [0:0]\n',
'-A INPUT -j RH-Firewall-1-INPUT\n',
'-A FORWARD -j RH-Firewall-1-INPUT\n',
'-A RH-Firewall-1-INPUT -i lo -j ACCEPT\n',
_trusted,
'-A RH-Firewall-1-INPUT -p icmp --icmp-type any -j ACCEPT\n',
'-A RH-Firewall-1-INPUT -p 50 -j ACCEPT\n',
'-A RH-Firewall-1-INPUT -p 51 -j ACCEPT\n',
'-A RH-Firewall-1-INPUT -p udp --dport 5353 -d 224.0.0.251 -j ACCEPT\n',
'-A RH-Firewall-1-INPUT -p udp -m udp --dport 631 -j ACCEPT\n',
'-A RH-Firewall-1-INPUT -m state --state ESTABLISHED,RELATED -j ACCEPT\n',
_open_ports,
'-A RH-Firewall-1-INPUT -j REJECT --reject-with icmp-host-prohibited\n',
'COMMIT\n' ]
create_file(buildroot, "/etc/sysconfig/iptables", content)
# enable firewall
if ks["firewall"].has_key("enabled"):
if run_script("/sbin/chkconfig iptables on", buildroot) != 0:
log.error("Could not enable firewall.")
else:
# use lokkit to configure firewall
fwargs = [ ]
if ks["firewall"].has_key("enabled"):
fwargs.append("--enabled")
if ks["firewall"].has_key("disabled"):
fwargs.append("--disabled")
if ks["firewall"].has_key("trusted"):
for trusted in ks["firewall"]["trusted"]:
fwargs.append("--trusted=%s" % trusted)
if ks["firewall"].has_key("ports"):
for port in ks["firewall"]["ports"]:
fwargs.append("--port=%s:%s" % (port[0], port[1]))
lokkit = "/usr/sbin/lokkit --quiet --nostart -f %s" % \
" ".join(fwargs)
if run_script(lokkit, buildroot) != 0:
log.error("Configuration of firewall failed.")
create_file(buildroot, "/etc/sysconfig/system-config-securitylevel",
[ '# Configuration file for system-config-securitylevel\n',
"\n",
"%s" % "\n".join(fwargs) ])
def network_config(ks, buildroot):
if not os.path.exists(buildroot+"/etc/sysconfig/network-scripts"):
os.mkdir(buildroot+"/etc/sysconfig/network-scripts")
# generate loopback network configuration if it does not exist
if not os.path.exists(buildroot+\
"/etc/sysconfig/network-scripts/ifcfg-lo"):
log.info1("Adding missing /etc/sysconfig/network-scripts/ifcfg-lo.")
create_file(buildroot, "/etc/sysconfig/network-scripts/ifcfg-lo",
[ 'DEVICE=lo\n',
'IPADDR=127.0.0.1\n',
'NETMASK=255.0.0.0\n',
'NETWORK=127.0.0.0\n',
"# If you're having problems with gated making 127.0.0.0/8 a martian,\n",
"# you can change this to something else (255.255.255.255, for example)\n",
'BROADCAST=127.255.255.255\n',
'ONBOOT=yes\n',
'NAME=loopback\n' ])
_hostname = None
_gateway = None
if ks.has_key("network") and len(ks["network"]) > 0:
# check network devices and set device for entries where no device
# is specified
network_devices = [ ]
for net in ks["network"]:
if net.has_key("device"):
if net["device"] in network_devices:
log.warning("'%s' is not unique.", net["device"])
else:
network_devices.append(net["device"])
for net in ks["network"]:
# get device or next free device
device = None
if net.has_key("device"):
device = net["device"]
if not device:
i = 0
device = "eth%d" % i
while device in network_devices:
i += 1
device = "eth%d" % i
net["device"] = device
for net in ks["network"]:
if not _hostname and net.has_key("hostname"):
_hostname = 'HOSTNAME=%s\n' % net["hostname"]
if not _gateway and net.has_key("gateway"):
_gateway = 'GATEWAY=%s\n' % net["gateway"]
device = net["device"]
if device[:3] == "ctc":
type = "CTC"
elif device[:4] == "iucv":
type = "IUCV"
elif device[:2] == "tr":
type = '"Token Ring"'
else:
type = "Ethernet"
bootproto = "none"
if net["bootproto"] and net["bootproto"] != "static":
bootproto = net["bootproto"]
try:
fd = open(buildroot + \
"/etc/sysconfig/network-scripts/ifcfg-%s" % \
device, "w")
except Exception, msg:
log.error("Configuration of '/etc/sysconfig/network-scripts/"
"ifcfg-%s' failed: %s", device, msg)
else:
fd.write('DEVICE=%s\n' % device)
fd.write('BOOTPROTO=%s\n' % bootproto)
if net.has_key("gateway"):
fd.write('GATEWAY=%s\n' % net["gateway"])
if net.has_key("netmask"):
fd.write('NETMASK=%s\n' % net["netmask"])
if net.has_key("ip"):
fd.write('IPADDR=%s\n' % net["ip"])
if net.has_key("essid"):
fd.write('ESSID=%s\n' % net["essid"])
if net.has_key("ethtool"):
fd.write('ETHTOOL_OPTS=%s\n' % net["ethtool"])
if net.has_key("class"):
fd.write('DHCP_CLASSID=%s\n' % net["class"])
if net.has_key("onboot"):
fd.write('ONBOOT=%s\n' % net["onboot"])
else:
fd.write('ONBOOT=yes\n')
fd.write('TYPE=%s\n' % type)
fd.close()
if net.has_key("wepkey"):
try:
fd = open(buildroot + \
"/etc/sysconfig/network-scripts/keys-%s" % \
device, "w")
except Exception, msg:
log.error("Configuration of '/etc/sysconfig/network-"
"scripts/keys-%s' failed: %s", device, msg)
else:
fd.write('KEY=%s\n' % net["wepkey"])
fd.close()
if net.has_key("class"):
fd.write('DHCP_CLASSID=%s\n' % net["class"])
if net.has_key("onboot"):
fd.write('ONBOOT=%s\n' % net["onboot"])
else:
fd.write('ONBOOT=yes\n')
fd.write('TYPE=%s\n' % type)
fd.close()
if net.has_key("wepkey"):
try:
fd = open(buildroot + \
"/etc/sysconfig/network-scripts/keys-%s" % \
device, "w")
except Exception, msg:
log.error("Configuration of '/etc/sysconfig/network-"
"scripts/keys-%s' failed: %s", device, msg)
else:
fd.write('KEY=%s\n' % net["wepkey"])
fd.close()
if not _hostname:
_hostname = 'HOSTNAME=localhost.localdomain\n'
if not _gateway:
_gateway = ""
# write /etc/sysconfig/network
create_file(buildroot, "/etc/sysconfig/network",
[ 'NETWORKING=yes\n', _hostname, _gateway ])
# vim:ts=4:sw=4:showmatch:expandtab
def network_config(ks, buildroot):
if not os.path.exists(buildroot + "/etc/sysconfig/network-scripts"):
os.mkdir(buildroot + "/etc/sysconfig/network-scripts")
# generate loopback network configuration if it does not exist
if not os.path.exists(buildroot+\
"/etc/sysconfig/network-scripts/ifcfg-lo"):
log.info1("Adding missing /etc/sysconfig/network-scripts/ifcfg-lo.")
create_file(buildroot, "/etc/sysconfig/network-scripts/ifcfg-lo", [
'DEVICE=lo\n', 'IPADDR=127.0.0.1\n', 'NETMASK=255.0.0.0\n',
'NETWORK=127.0.0.0\n',
"# If you're having problems with gated making 127.0.0.0/8 a martian,\n",
"# you can change this to something else (255.255.255.255, for example)\n",
'BROADCAST=127.255.255.255\n', 'ONBOOT=yes\n', 'NAME=loopback\n'
])
_hostname = None
_gateway = None
if ks.has_key("network") and len(ks["network"]) > 0:
# check network devices and set device for entries where no device
# is specified
network_devices = []
for net in ks["network"]:
if net.has_key("device"):
if net["device"] in network_devices:
log.warning("'%s' is not unique.", net["device"])
else:
network_devices.append(net["device"])
for net in ks["network"]:
# get device or next free device
device = None
if net.has_key("device"):
device = net["device"]
if not device:
i = 0
device = "eth%d" % i
while device in network_devices:
i += 1
device = "eth%d" % i
net["device"] = device
for net in ks["network"]:
if not _hostname and net.has_key("hostname"):
_hostname = 'HOSTNAME=%s\n' % net["hostname"]
if not _gateway and net.has_key("gateway"):
_gateway = 'GATEWAY=%s\n' % net["gateway"]
device = net["device"]
if device[:3] == "ctc":
type = "CTC"
elif device[:4] == "iucv":
type = "IUCV"
elif device[:2] == "tr":
type = '"Token Ring"'
else:
type = "Ethernet"
bootproto = "none"
if net["bootproto"] and net["bootproto"] != "static":
bootproto = net["bootproto"]
try:
fd = open(buildroot + \
"/etc/sysconfig/network-scripts/ifcfg-%s" % \
device, "w")
except Exception, msg:
log.error(
"Configuration of '/etc/sysconfig/network-scripts/"
"ifcfg-%s' failed: %s", device, msg)
else:
fd.write('DEVICE=%s\n' % device)
fd.write('BOOTPROTO=%s\n' % bootproto)
if net.has_key("gateway"):
fd.write('GATEWAY=%s\n' % net["gateway"])
if net.has_key("netmask"):
fd.write('NETMASK=%s\n' % net["netmask"])
if net.has_key("ip"):
fd.write('IPADDR=%s\n' % net["ip"])
if net.has_key("essid"):
fd.write('ESSID=%s\n' % net["essid"])
if net.has_key("ethtool"):
fd.write('ETHTOOL_OPTS=%s\n' % net["ethtool"])
if net.has_key("class"):
fd.write('DHCP_CLASSID=%s\n' % net["class"])
if net.has_key("onboot"):
fd.write('ONBOOT=%s\n' % net["onboot"])
else:
fd.write('ONBOOT=yes\n')
fd.write('TYPE=%s\n' % type)
fd.close()
if net.has_key("wepkey"):
try:
fd = open(buildroot + \
"/etc/sysconfig/network-scripts/keys-%s" % \
device, "w")
except Exception, msg:
log.error(
"Configuration of '/etc/sysconfig/network-"
"scripts/keys-%s' failed: %s", device, msg)
else:
fd.write('KEY=%s\n' % net["wepkey"])
fd.close()
if net.has_key("onboot"):
fd.write('ONBOOT=%s\n' % net["onboot"])
else:
fd.write('ONBOOT=yes\n')
fd.write('TYPE=%s\n' % type)
fd.close()
if net.has_key("wepkey"):
try:
fd = open(buildroot + \
"/etc/sysconfig/network-scripts/keys-%s" % \
device, "w")
except Exception, msg:
log.error(
"Configuration of '/etc/sysconfig/network-"
"scripts/keys-%s' failed: %s", device, msg)
else:
fd.write('KEY=%s\n' % net["wepkey"])
fd.close()
if not _hostname:
_hostname = 'HOSTNAME=localhost.localdomain\n'
if not _gateway:
_gateway = ""
# write /etc/sysconfig/network
create_file(buildroot, "/etc/sysconfig/network",
['NETWORKING=yes\n', _hostname, _gateway])
# vim:ts=4:sw=4:showmatch:expandtab
except IndexError:
print('Нет параметра')
elif command == 'go_back':
splited_path = path.split("/")
path = "/".join(splited_path[:-2]) + "/"
f = open("path.txt", 'w')
f.close()
with open("path.txt", "w") as file:
file.write(path)
elif command == 'create_file':
try:
name = sys.argv[2]
text = sys.argv[3]
except IndexError:
print('Нет параметра')
create_file(path, name, text)
elif command == 'create_folder':
try:
name = sys.argv[2]
except IndexError:
print('Нет параметра')
create_folder(path, name)
elif command == 'delete':
try:
name = sys.argv[2]
except IndexError:
print('Нет параметра')
delete(path + name)
elif command == 'copy':
try:
name = sys.argv[2]
if os.path.exists(image_filename):
print(image_filename + " exists")
print("reading:", image_filename)
actual = f.read_filenames_from_disk(image_filename)
else:
print(image_filename + " does not exist.")
print("examining camera folder for files")
good = get_filenames_from_camera_folders(path, root=root)
print(type(good), len(good), "good file names")
print(good[0])
print(type(good), len(good), "good file names")
#print(good[0])
ts = tuple(zip(*good))
(images, jsons, labels) = ts
p(images, jsons, labels)
print(labels[0])
if not (len(images) == len(jsons) and len(images) == len(labels)):
print("lengths of jsons and/or labels are not equal!")
#exit()
#filename=join(f.get_my_path(),"image_"+"good_file_names.txt")
f.create_file(image_filename, images)
f.create_file(json_filename, jsons)
f.create_file(label_filename, labels)
print("reading:", image_filename)
actual = f.read_filenames_from_disk(image_filename)
print("check")
if not f.is_ok(actual, images):
print("badness")
print(len(actual), "image file names")
# make into a function
