Познакомьтесь с пентестом веб-приложений на практике в нашем новом бесплатном курсе
from os import stat, path, walk, listdir
from pathlib import Path
from sys import exit as ex
def correct_size(bts: int, ending='iB') -> str:
"""
Корректируется размер файла.
:param bts: размер файла, целое число.
:param ending: суффикс, добавляемый к метрике.
:return: строка, корректированный размер файла.
"""
size = 1024
for item in ["", "K", "M", "G", "T", "P"]:
if bts < size:
return f"{bts:.2f} {item}{ending}"
bts /= size
def folder_scan(path_dir: str) -> str:
"""
Определение размера директории. Сканируется директория.
Определяется размер каждого файла находящегося в ней.
Суммируется в общей переменной. После завершения сканирования
возвращается скорректированный размер.
:param path_dir: строка, путь к директории для сканирования.
:return: строка, скорректированный размер файла.
"""
file_size = stat(path_dir).st_size
for root, dirs, files in walk(path_dir):
for file in files:
file_size = file_size + stat(path.join(root, file)).st_size
return correct_size(file_size)
def list_dir(path_dir: str):
"""
Сканирование директории. Определение размера файлов и папок.
Добавление данных об имени и размере в текстовый документ.
Сортировка полученных объектов на файлы и папки в процессе
итерации по ним. Если файл, определяем размер сразу. Если папка,
передаем путь к ней в функцию получения размера папки.
:param path_dir: строка, путь к директории для сканирования.
"""
rez = sorted(listdir(path_dir))
file_name = f'{Path(path_dir).name}'
with open(f"{file_name}.txt", "w") as file:
for n, item in enumerate(rez):
if path.isdir(path.join(path_dir, item)):
size = folder_scan(path.join(path_dir, item))
else:
size = correct_size(stat(path.join(path_dir, item)).st_size)
file.write(f"{n + 1} {item} - {size}\n")
print(f'Список файлов смотрим здесь - {file_name}.txt ')
def main():
"""
Запрос у пользователя пути к сканируемой директории.
Запуск функции сканирования с передачей в нее
полученного от пользователя пути.
"""
path_dir = input("Введите путь к сканируемой директории: ")
if not path.exists(path_dir):
print("Введенной вами директории не существует.")
ex(0)
list_dir(path_dir)
if __name__ == "__main__":
main()Опять ошибк:
path = 'E:\Мой Python\__Видеокурсы Python' нужно либо экранировать, либо поставить букву r, чтобы не обрабатывались обратные слеши.path = r'E:\Мой Python\__Видеокурсы Python'. Это в том случае, если не передавать путь к папке с помощью input, а передавать в коде.Спасибо. В моём исходном скрипте всё прввильно.
Спасибо. Вы отлично разбираетесь в питоне. Мне тоже Вам хочется быть полезным.
Ещё у меня проблемка.
# -*- coding: utf-8 -*
import glob
dirname = r'C:\Users\Пк\Desktop\1_Изучаем Python'
line = 'Яндекс Лицей'
for name in glob.glob(f"{dirname}/**/*.m3u"):
file_info = open(name, mode="r", encoding='utf-8').read()
if line in fileinfo:
print(f'Нашел нужную строчку в файле {name}')
# 1_Изучаем Python - это моя папка
# C:\Users\Пк\Desktop\1_Изучаем Python - это путь к папке
# Яндекс Лицей - это ищем в файлах с расширением .m3uПапка путь: C:\Users\Пк\Desktop\2_Мои IPTV
glob.glob(f"{dirname}/**/*.m3u")
import sys
from glob import glob
from pathlib import Path
def scan(path_dir: str, req: str):
"""
Выполняется итерация по каталогам(рекурсивно) и ищется
файл по заданному шаблону. После того как файл найден,
он открывается для чтения.
Выполняется условие, по поиску вхождения в текст файла.
Если найден, печатаем где. Если нет, двигаемся дальше.
:param path_dir: строка, путь к директории для сканирования.
:param req: строка, слово или предложение, вхождение которого нужно найти.
"""
try:
print("")
for name in glob(str(Path(path_dir) / "**" / "*.m3u"), recursive=True):
with open(name, "r", encoding='utf-8') as resp:
if req in resp.read():
print(f"Найдено совпадение в файле: {name}")
except KeyboardInterrupt:
print("\nGood By!")
def main():
try:
path_dir = input("Введите путь к директории для сканирования: ")
req = input("Введите искомое слово или словосочетание: ")
if not Path(path_dir).exists():
print("Введенного пути не существует!")
sys.exit(0)
scan(path_dir=path_dir, req=req)
except KeyboardInterrupt:
print("\nGood By!")
if __name__ == "__main__":
main()import sys
from glob import glob
from pathlib import Path
def scan(path_dir: str, req: str):
"""
Выполняется итерация по каталогам(рекурсивно) и ищется
файл по заданному шаблону. После того как файл найден,
он открывается для чтения построчно. В цикле выполняется
итерация по каждой строке, и проверяется вхождение
искомого слова или предложения.
Если вхождение найдено, добавляем строку с номером и текстом
во временный список.
После того как цикл завершиться, проверяем какова длина
списка. Если она больше нуля, печатаем сколько совпадений
и выводим каждую строку, в которой найдено совпадение.
:param path_dir: строка, путь к директории для сканирования.
:param req: строка, слово или предложение, вхождение которого нужно найти.
"""
try:
for name in glob(str(Path(path_dir) / "**" / "*.m3u"), recursive=True):
lines = []
with open(name, "r", encoding='utf-8') as resp:
for n, line in enumerate(resp.readlines()):
if req in line:
lines.append(f"Строка: {n+1}, {line.strip()}")
if len(lines) > 0:
print(f'\n[+] Совпадений: {len(lines)}, в файле: {name}')
for ln in lines:
print(f" - {ln}")
except KeyboardInterrupt:
print("\nGood By!")
def main():
try:
path_dir = input("Введите путь к директории для сканирования: ")
req = input("Введите искомое слово или словосочетание: ")
if not Path(path_dir).exists():
print("Введенного пути не существует!")
sys.exit(0)
scan(path_dir=path_dir, req=req)
except KeyboardInterrupt:
print("\nGood By!")
if __name__ == "__main__":
main()
Спасибо огромное. Вы мне целую лекция прочитали. Всё отлично работает. Впервые получаю такю грамотную поддержку.
Обучение наступательной кибербезопасности в игровой форме. Начать игру!
