Skip to content

Commit 34fd1ee

Browse files
committed
Edited file and link deletion when exist (switched to oe_common.rm)
1 parent be279e1 commit 34fd1ee

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

ovtp/server/ovtp_server.py

+7-4
Original file line numberDiff line numberDiff line change
@@ -116,10 +116,12 @@ async def handle_packet(self):
116116
else:
117117
self.aes = ov_aes_cipher.AESCipher('pass')
118118

119-
if data_type == b'file' and os.path.isfile(filename):
120-
os.remove(filename)
121-
elif data_type == b'link' and os.path.isfile(filename):
122-
os.remove(filename)
119+
if data_type == b'file' and os.path.exists(filename):
120+
print(f'File exists, deleting: "{filename}"')
121+
oe_common.rm(filename)
122+
elif data_type == b'link' and os.path.exists(filename):
123+
print(f'Link exists, deleting: "{filename}"')
124+
oe_common.rm(filename)
123125

124126
if address in self.server.saved_keys:
125127
ov_sign = ovcrypt.OvSign(self.server.saved_keys[address].key)
@@ -356,6 +358,7 @@ async def handle_packet(self):
356358
print(f'Received "{filename}"')
357359
if not os.path.isfile(filename):
358360
print(f'File not exist, creating: "{filename}"')
361+
oe_common.check_create_dir(filename)
359362
pathlib.Path(filename).touch()
360363
elif data_type == b'link':
361364
data = b''.join(data_parts)

setup.cfg

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[metadata]
22
name = ovtp
3-
version = 1.0.6.1
3+
version = 1.0.6.2
44
author = Dmitry Yakovlev
55
author_email = [email protected]
66
description = Over Transfer Protocol

0 commit comments

Comments
 (0)