Открыть файл по расширению

[mrprim]

Доброго времени суток, снова.
зачастил я к вам с вопросами, но что уж поделать если есть люди которые умело поясняют.
В общем, такая ситуация, есть некий текстовый файл к примеру: 1234.uuu
так вот, в нём хранится информация, у меня есть метод который берет эту информацию и закидывает в файл *.ooo
Мне нужно написать метод который сам глянет все файлы .uuu и создаст файлы с таким же именем но .ooo

[pompiduskus]

Можно решить эту проблему по разному.
Самый просто наврное будет такой:

Я пишу из под Linux,
поэтому некоторые вещи могуть быть не такими, если ты пишешь из под окон.
Ты под какой OS пишешь ?

Если посмотреть на список файлов в моем случае, выглядит так:
Есть файлы которые мне нужны, и есть которые я не хочу обрабатывать.

PHP код:

toor@tr:/tmp/data-dir$ ls -l
A.uuu
A.wrong.file
B.uuu
B.wrong.file
main.py 


Первое что нужно вообще получить (программно), это список файлов.

Можно решить так:

PHP код:


#!/usr/bin/python
import os

# =======================================================
CURR_DIR = "/tmp/data-dir/";

ALL_FILES = os.listdir( CURR_DIR );

for _file in ALL_FILES:

    print("PATH: ["+CURR_DIR+"] FILE: ["+_file+"]" ); 


На выходе подучим вот такое.

toor@tr:/tmp/data-dir$ python main.py 
PATH: [/tmp/data-dir/] FILE: [B.wrong.file]
PATH: [/tmp/data-dir/] FILE: [B.uuu]
PATH: [/tmp/data-dir/] FILE: [A.wrong.file]
PATH: [/tmp/data-dir/] FILE: [main.py]
PATH: [/tmp/data-dir/] FILE: [A.uuu]

Теперь нужно определиться, как мы вообще будет фильтровать эти фалы.
Берем все файлы которые имеют разширение "*.uuu"

Что-бы взять разширение фала, можно просто его разбить на части по (.) точке.

Или воспользоваться встроенным решение Питона.

Я привиду оба решения, сам решай что тебе больше нравится:
Но по сути они делают одно и тоже, разница будет лишь в имплементации.
Что соответственно скажется на скорости, и прожорливости памяти.

На выходе подучим вот такое. (1ом и 2ом варианте)
За мальким исключение: вот этот метод "os.path.splitext"

запишет разширение вот так ".exe", а первый так "exe"

PHP код:


for _file in ALL_FILES:

    F_DATA = _file.split(".");
    print("F_NAME: ["+F_DATA[0]+"] F_EXE: ["+F_DATA[1]+"]");

    F_NAME, F_EXE = os.path.splitext( CURR_DIR+_file );
    print("F_NAME: ["+F_NAME+"] F_EXE: ["+F_EXE+"]"); 


toor@tr:/tmp/data-dir$ python main.py 
F_NAME: [B] F_EXE: [wrong]
F_NAME: [B] F_EXE: [uuu]
F_NAME: [A] F_EXE: [wrong]
F_NAME: [main] F_EXE: [py]
F_NAME: [A] F_EXE: [uuu]

Теперь осталось только работать с фалами, но это ты сам знаешь что там и как ты делашь.

Теперь все в месте + создание / запись новых фалов.

PHP код:

#!/usr/bin/python
import os

# =======================================================
CURR_DIR = "/tmp/data-dir/";

ALL_FILES = os.listdir( CURR_DIR );

for _file in ALL_FILES:

    # Разбиваем строку по [.] и получаем масив строк
    F_DATA = _file.split(".");

    F_NAME = F_DATA[0]; # Имя файла
    F_EXE  = F_DATA[1]; # Разширение файла

    if F_EXE == "uuu":

        # Считываем по строчно файл [CURR_DIR+_file]
        # И добавляем каждую строку в массив 
        DATA_FROM_FILE = open( CURR_DIR+_file ).readlines(); 

        # Открываем новый файл для записи "w == запись"
        FS_OUT = open( F_NAME+".ooo", "w" );

        for data in DATA_FROM_FILE:
            FS_OUT.write( data+"\n" );

        # Закрываем файл
        FS_OUT.close();

# ======================================================= 


Есть одно но: Если вдруг на попядется файл который не будет иметь разширения, то программа упадет.

Что-бы обезопасить себя, можно делать проверку.

PHP код:


#!/usr/bin/python
import os

# =======================================================
CURR_DIR = "/tmp/data-dir/";

ALL_FILES = os.listdir( CURR_DIR );

for _file in ALL_FILES:

    # Разбиваем строку по [.] и получаем масив строк
    F_DATA = _file.split(".");


    try:

        F_NAME = F_DATA[0]; # Имя файла
        F_EXE  = F_DATA[1]; # Разширение файла

    except Exception as _err:

        #print( "ERROR: "+str(_err) );
        # Сразу переходим на следующую итерацию
        # Так-как это 100% не тот файл котрые нам нужен 
        continue;


    if F_EXE == "uuu":

        # Считываем по строчно файл [CURR_DIR+_file]
        # И добавляем каждую строку в массив 
        DATA_FROM_FILE = open( CURR_DIR+_file ).readlines(); 

        # Открываем новый файл для записи "w == запись"
        FS_OUT = open( F_NAME+".ooo", "w" );

        for data in DATA_FROM_FILE:
            FS_OUT.write( data+"\n" );

        # Закрываем файл
        FS_OUT.close();

# ======================================================= 


И последнее, если файл в который мы будем писать существует.

А мы открываем его с фдагом "w"

fs = open("file_name", "w"); Файл будет полностью стерт, и перезаписан.

[mrprim]

Сообщение от pompiduskus

Можно решить эту проблему по разному.
Самый просто наврное будет такой:

Я пишу из под Linux,
поэтому некоторые вещи могуть быть не такими, если ты пишешь из под окон.
Ты под какой OS пишешь ?
...

И последнее, если файл в который мы будем писать существует.

А мы открываем его с фдагом "w"

fs = open("file_name", "w"); Файл будет полностью стерт, и перезаписан.

всё супер, изменил под себя, получилось то что нужно.
А то что файл перезаписывается это еще лучше
Очередной раз спасибо вам!
Питон используется в проекте и изучать его с самого начала не хватает времени и о многих встроенных фичах питона не знал. Очень хорошо объясняете!=)

[pompiduskus]

Вот и хорошо! Не зря старался.
Ты главное не просто копируй, вникай в то что происходит.

[mrprim]

Сообщение от pompiduskus

Вот и хорошо! Не зря старался.
Ты главное не просто копируй, вникай в то что происходит.

а как сделать если нужно ввести имя файла самому и с ним провести те же операции, пробую:

PHP код:

s=raw_input("Enter filename: ")
str = s
for word in str:
        FILE_DATA = word.split(".")
        print FILE_DATA 


при вводе имени файла 123.uuu
выводит посимвольно [1] [2] [3] [u] [u] [u]

если делаю так:

PHP код:


s=raw_input("Enter filename: ")
str = s
for word in str.split("."):
        print word 


при вводе 123.uuu
выводит 2 элемента [123] [uuu]
но если пытаюсь получить доступ ко второму элементу word[1] То выводит просто по второму символу от каждого элемента

[mrprim]

всегда бесило, если проблема решилась то человек форум закрывает, поэтому если кто то забредёт в эту тему по той же проблеме, я сделал так:

PHP код:

CURR_DIR = "/MyBasicPy/"
ALL_FILES = os.listdir( CURR_DIR )

        while 1:
                i=0
                for word in s.split("."):
                        if i==0:
                                FILE_NAME = word
                        else: FILE_EXE = word
                        i+=1
                for _file in ALL_FILES:
                        FILE_DATA_DIRR = _file.split(".")
                        try: 
                                FILE_NAME_DIRR = FILE_DATA_DIRR[0]
                                FILE_EXE_DIRR  = FILE_DATA_DIRR[1]
                        except Exception as _err:  
                                continue
                        if (FILE_NAME_DIRR == FILE_NAME) and (FILE_EXE_DIRR == "uuu"):
                                print "FILE: ["+_file+"]"
                                f=open(_file, "r")
                                save_hex(FILE_NAME_DIRR+".ooo") //сторонний метод для перезаписи в другой файл
                                print ("The bytecode has been written to the "+FILE_NAME_DIRR+".ooo")
                                FILE_EXE = 0
                                s=0
                        else:
                                continue
                if s=='ss':
                        print ("Thanks, bye!")
                        break 


если кто может подсказать как исправить эту кривую часть

PHP код:

                i=0
                for word in s.split("."):
                        if i==0:
                                FILE_NAME = word
                        else: FILE_EXE = word
                        i+=1 


то буду рад это сделать)

[pompiduskus]

Все потому-что ты не вникаешь в суть, а просто копируешь и все. Вот если просто посмотреть сюда

Ты считываешь имя файла, а потом начинаешь перебирать его как будто-бы это массив. Но там строка которую ты-же ему и передал.

Но Питон умный, он делает все что может. Вот и получается что он начинает делить строку на символы.

А вот так будет то что ты хотел, наверное.

PHP код:


_input = raw_input("Enter filename: ");
print( _input );

FILE_DATA = _input.split(".");

FILE_NAME = FILE_DATA[0];
FILE_EXE   = FILE_DATA[1];