lab2_python/task3.py at main · d7anSeR/lab2_python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import csv
import os
import shutil
from random import randint


def create_dir(dir_name: str) -> None:
    '''The function of creating a new directory with the specified name dir_name'''
    name_dir_folder = os.path.join(dir_name, "dataset")
    if not os.path.isdir(dir_name):
        os.mkdir(dir_name)
    if not os.path.isdir(name_dir_folder):
        name = os.path.join(dir_name, "dataset")
        os.mkdir(name)


def copy_dir(dir_name: str, good_name: str, bad_name: str, annotation_name: str) -> None:
    '''
    the function copies the dataset to a
    new directory with a random file name by means of creating this directory
    using the dir_create function and writes the absolute path,
    relative path and class name to a new csv file to be able
    to define the instance class
    '''
    create_dir(dir_name)
    good = os.path.join("dataset", good_name)
    bad = os.path.join("dataset", bad_name)
    list_good = os.listdir(good)
    list_bad = os.listdir(bad)
    list_tmp = []
    for file_dataset in os.listdir("dataset"):
        path_p = os.path.join("dataset", file_dataset)
        class_name = os.listdir(path_p)
        for elem in class_name:
            flag = True
            number = randint(0, 10000)
            while flag:
                if number in list_tmp:
                    number = randint(0, 10000)
                else:
                    list_tmp.append(number)
                    flag = False
            old_file = os.path.join(path_p, elem)
            new_file = os.path.join(
                dir_name, "dataset", f"{number}.txt")
            shutil.copy(old_file, new_file)
            with open(annotation_name, mode="a", encoding="UTF-16", newline='') as f:
                writer = csv.writer(f, delimiter=';')
                result_file = os.path.join("dataset", f"{number}.txt")
                abspath_f = os.path.join(
                    os.path.abspath(dir_name), result_file)
                otnos = os.path.join(dir_name, result_file)
                writer.writerow([abspath_f, otnos, file_dataset])
            pass
    print("3 task completed")


def run3(dir_name: str, good_name: str, bad_name: str, annotation_name: str) -> None:
    '''The function of launching a function that copy the dataset to a new directory with a random file name'''
    copy_dir(dir_name, good_name, bad_name, annotation_name)