def setUp(self):
self.output_connections = [1, 2, 3, 4, 5]
# delete pre-existing output data
for connection_num in self.output_connections:
deleteFile(outputFilename(connection_num))
# create a pandas dataframe to write out
self.data = read("#1")
def setUp(self):
self.filepath = 'delete_test_1.txt'
# delete the file to start with clean slate
deleteFile(self.filepath)
# create new file in its place
with open(self.filepath, 'w') as file:
file.write('test')
try:
fileExists(self.filepath, throw_error=True)
except:
self.fail("deleteFile() raised an error unexpectedly!")
def __exit__(self, type, value, traceback):
self.closeConnection()
# if temporary file, attempt to clean up on exit
if self.temporary:
try:
deleteFile(self.filepath, debug=self.debug)
except:
# (this is not the end of the world)
if self.debug:
print("".join([
"Unable to delete temp file (will be cleanded up later",
"by asset manager) -- {}".format(self.filepath),
]))
def __returnConnection(self):
error_msg = "Unable to connect to input data"
try:
# if file exists, and not creating a new db, then throw error
if self.create_new:
if os.path.isfile(self.filepath):
try:
deleteFile(self.filepath, debug=self.debug)
except:
raise PermissionError(
"unable to delete file: {}".format(self.filepath))
if not os.access(os.path.dirname(self.filepath), os.W_OK):
raise PermissionError(
"unable to write to filepath: {}".format(
self.filepath))
return None
elif fileExists(self.filepath,
throw_error=not (self.create_new),
msg=error_msg):
# open connection and attempt to read one row
# to confirm that it is a valid file
if self.fileformat.filetype == "sqlite":
connection = sqlite3.connect(self.filepath)
connection.execute("select * from sqlite_master limit 1")
elif self.fileformat.filetype == "yxdb":
connection = pyxdb.AlteryxYXDB()
connection.open(self.filepath)
# test that its a real yxdb file
connection.get_num_records()
else:
self.__formatNotSupportedYet()
return connection
except:
try:
connection.close()
except:
pass
raise ConnectionError(fileErrorMsg(error_msg, self.filepath))
def testWriteDeleteWriteDelete(self):
# delete the file to start with clean slate
deleteFile(self.filepath)
# does file exist?
result = fileExists(self.filepath)
self.assertFalse(result)
def tearDown(self):
deleteFile(self.filename)
def setUp(self):
self.connection = 5
self.filename = outputFilename(self.connection)
self.data = read("#2")
deleteFile(self.filename)
self.test = True
def tearDown(self):
for connection_num in self.output_connections:
deleteFile(outputFilename(connection_num))