Commit bb3082d88c6ecbe64bb71b07fa50bb06269e8a04
1 parent
0d6fbe51
Exists in
master
Fix #2. Check prefix before add or remove prefix.
Showing
1 changed file
with
11 additions
and
5 deletions
Show diff stats
src/files.py
| @@ -36,8 +36,11 @@ class PireneaFiles(object): | @@ -36,8 +36,11 @@ class PireneaFiles(object): | ||
| 36 | prefix += "_" | 36 | prefix += "_" |
| 37 | for path, _dirs, files in os.walk(self.folder): | 37 | for path, _dirs, files in os.walk(self.folder): |
| 38 | for filename in files: | 38 | for filename in files: |
| 39 | - new_filename = prefix + filename | ||
| 40 | - os.rename(os.path.join(path, filename), os.path.join(path, new_filename)) | 39 | + if filename.startswith("P"): |
| 40 | + print("File %s has already a prefix.") | ||
| 41 | + else: | ||
| 42 | + new_filename = prefix + filename | ||
| 43 | + os.rename(os.path.join(path, filename), os.path.join(path, new_filename)) | ||
| 41 | 44 | ||
| 42 | def remove_prefix(self, prefix="P0"): | 45 | def remove_prefix(self, prefix="P0"): |
| 43 | """ | 46 | """ |
| @@ -47,9 +50,12 @@ class PireneaFiles(object): | @@ -47,9 +50,12 @@ class PireneaFiles(object): | ||
| 47 | prefix += "_" | 50 | prefix += "_" |
| 48 | for path, _dirs, files in os.walk(self.folder): | 51 | for path, _dirs, files in os.walk(self.folder): |
| 49 | for filename in files: | 52 | for filename in files: |
| 50 | - f = filename.split(prefix) | ||
| 51 | - new_filename = prefix.join(f[1:]) | ||
| 52 | - os.rename(os.path.join(path, filename), os.path.join(path, new_filename)) | 53 | + if filename.startswith(prefix): |
| 54 | + f = filename.split(prefix) | ||
| 55 | + new_filename = prefix.join(f[1:]) | ||
| 56 | + os.rename(os.path.join(path, filename), os.path.join(path, new_filename)) | ||
| 57 | + else: | ||
| 58 | + print("File {0} does not begin with prefix : {1}".format(filename, prefix[:2])) | ||
| 53 | 59 | ||
| 54 | def __check_prefix(self, prefix="P0"): | 60 | def __check_prefix(self, prefix="P0"): |
| 55 | """ | 61 | """ |