PKFFv5.2.1.py

# The following code is written by Rohit Saxena. STUDY PURPOSE ONLY. Any unauthorized use or modification is prohibited.
import ctypes
import datetime
import ftplib
import glob
import os
import shutil
import string
from urllib.error import URLError
from urllib.request import *
import docx
import openpyxl
import logging
import threading
import time
from tqdm import tqdm

# import PyPDF2
# import re

errorCount, i, j = 0, 0, 0
rvs = []
host = ""  # FTP Host URL
ftp_id = ""  # FTP CLIENT ID
ftp_pw = ""  # FTP CLIENT PASSWORD
_ftp_root_folder_ = 'htdocs'  # FTP root folder
newdir = os.environ['COMPUTERNAME']  # FTP Main folder name
ft = []
todaysDate = str(datetime.datetime.today())  # FTP sub-folder name
ftp = ftplib.FTP()


def thread_function(name):
    rstqdm = tqdm(range(100))
    time.sleep(2)
    # logging.info("Scan started %s: loading...", name)
    for i in rstqdm:
        time.sleep(1)
        if i == 100:
            rstqdm.clear()
            logging.info("Please wait scan is in progress...")


def _rS_sign():
    print("\n><----------------------------------------------><\n")
    print("|""---------""    ""------------")
    print("|""        ""|""  ""  |")
    print("|""        ""|""  ""  |")
    print("|""---------""    ""|")
    print("|""-""            ""------------")
    print("|""  ""-""                     ""|")
    print("|""    ""-""                   ""|")
    print("|""       ""-""   ""  ------------")
    print("The following code is written by Rohit Saxena. STUDY PURPOSE ONLY. Any unauthorized use or modification is "
          "prohibited.")
    print("\n><----------------------------------------------><\n")
    return True


def _rSAdmin_Check():
    try:
        is_admin = (os.getuid() == 0)
    except AttributeError:
        is_admin = ctypes.windll.shell32.IsUserAnAdmin() != 0
    return is_admin


def _rSInternet_Check_():  # Internet Tester
    while True:
        try:
            urlopen('https://www.central16.in', timeout=1)
            return
        except URLError as e:
            print('Internet Connectivity Check --> URL Error: ', e.reason)  # remove print is building exe
            pass


def _RsBackupWriter(rrs, pp):  # local backup (unfinished)
    global j
    print('BACKUP FOLDER STARTED \nTotal File address written : ', len(rrs))
    trs = open(pp, "a")
    for item in rrs:
        trs.write("%s\n" % item)
    print('Backup File Created')
    file1 = open(pp, "r")
    print(file1.read())
    file1.close()


def _RsXlsx_(SS, CC):  # XLSX Reader
    rs = openpyxl.load_workbook(CC, read_only=True, data_only=True)
    ws = rs.active
    for i in range(1, ws.max_row + 1):
        for j in range(1, ws.max_column + 1):
            if SS == ws.cell(i, j).value:
                print("Found in cell :", ws.cell(i, j), "File Location :", CC)
                rvs.append(CC)


'''
def _rsPDF_(SS, CC):  # PDF reader

    filepdf = PyPDF2.PdfFileReader(CC)
    NumPages = filepdf.getNumPages()
    for i in range(0, NumPages):
        PageObj = filepdf.getPage(i)
        Text = PageObj.extractText()
        ResSearch = re.search(SS, Text)
        print("-------------> ", ResSearch)
        if ResSearch in filepdf:
            print('String', ResSearch, 'File Path: ', CC)
            rvs.append()
'''


def _RsTxt_(SS, CC):  # TXT Reader
    file1 = open(CC, 'r', encoding='UTF8', errors='ignore')
    flag = 0
    index = 0
    for line in file1:
        index += 1
        if SS in line.lower():
            flag = 1
            break
    if flag != 0:
        print('String', SS, 'Found In Line: ', index, 'File Path: ', CC)
        rvs.append(CC)
    file1.close()


def _RsDocx_(SS, CC):  # DOCX Reader
    flag = 0
    index = 0
    doc = docx.Document(CC)
    for para in doc.paragraphs:
        index += 1
        if SS in para.text:
            flag = 1
            break
    if flag != 0:
        print('String', SS, 'Found In Line: ', index, 'File Path: ', CC)
        rvs.append(CC)


def _RsDirCheck_(rsdc):  # FTP Main Folder Checker
    x = 0
    for g in rsdc:
        if newdir in g:
            x = 1
    if x == 0:
        print(" New Directory created ---> ", newdir)
        ftp.mkd(newdir)


def _RsTodayFolderCheck_(rstfc):  # FTP Sub-Folder Checker
    x = 0
    for g in rstfc:
        if todaysDate in g:
            x = 1
    if x == 0:
        ftp.mkd(todaysDate)
        print("New Today's date folder created ---> ", todaysDate)


def _Rs_ftp_COPY_():  # FTP Client connector
    print('uploading data...')
    port = 21
    ftp.connect(host, port)
    print(ftp.getwelcome())
    print("Logging in...")
    ftp.login(ftp_id, ftp_pw)
    ftp.cwd(_ftp_root_folder_)
    ftp.retrlines('LIST', ft.append)
    _RsDirCheck_(ft)
    ftp.cwd(newdir)
    ft.clear()
    ftp.retrlines('LIST', ft.append)
    _RsTodayFolderCheck_(ft)
    ftp.cwd(todaysDate)


def _rS_FTP_file_transf(rvs):  # FTP File uploader
    try:
        rs = 0
        print('Uploading Data...')
        while rs != len(rvs):
            filename = rvs[rs]
            file_name, file_extension = os.path.splitext(filename)
            _FTP_File_Name_ = os.path.basename(file_name) + file_extension
            _ftpNewFileNMCmd_ = "STOR %s" % _FTP_File_Name_
            with open(filename, "rb") as file:
                ftp.storbinary(_ftpNewFileNMCmd_, file)
        rs += 1
        print('Upload Complete !')
        ftp.dir()
    except Exception:
        print(Exception)


def main():
    global errorCount
    if _rSAdmin_Check():
        os.system('color 2')
        print("<-----Admin Access----->")
    else:
        os.system('color 4')
        print("<-----No Admin Access----->")
    global i, j
    global rvs
    #    val = input("Enter extension : ")    #If specific extension only required uncheck and add val in _Rel_Path
    ff = input("Enter string : ").lower()

    rs = ['%s:\\' % d for d in string.ascii_uppercase if os.path.exists('%s:\\' % d)]
    print('\n><----------------------------------------------><\nGENERAL DATA\nActive local drives ---> ', rs)
    FtypeXLSX = '.xlsx'
    FtypeTXT = '.txt'
    FtypeDOCX = '.docx'
    FtypePDF = '.pdf'
    _Rel_Exp_ = '**/*'
    _Rel_Path_ = _Rel_Exp_
    # print('Selected file type = ', val)
    rss = []

    bck_dir = "RSSF"  # LINE 154 - 166 REFERS LOCAL BACKUP (UNFINISHED)
    bck_file = "RSSF.txt"
    print('Reuslt folder named RSSF is created in first local drive from available drive list')
    pth = os.path.join(rs[0], bck_dir)
    if os.path.exists(pth):  # this if will flush the old backup folder and re-create new backup folder
        shutil.rmtree(pth)
    os.makedirs(pth)
    print('Result folder path: ', pth)
    print('Result file RSSF.txt is created in hidden folder rss')
    CN = os.path.join(pth, bck_file)
    print('Result file location: ', CN)
    file1 = open(CN, "a")
    file1.close()

    format = "%(asctime)s: %(message)s"
    logging.basicConfig(format=format, level=logging.INFO,
                        datefmt="%H:%M:%S")
    x = threading.Thread(target=thread_function, args=(1,))
    x.start()

    if _rS_sign():
        print('Below files are present in system: '"\n")
        while i != len(rs):
            os.chdir(rs[i])
            for file in glob.glob(_Rel_Path_, recursive=True):
                # print(file)
                completeName = os.path.join(rs[i], file)
                # print('File path: ', completeName)
                file_name, file_extension = os.path.splitext(completeName)
                if file_extension == FtypeXLSX:
                    try:
                        _RsXlsx_(ff, completeName)
                    except Exception:
                        errorCount += 1
                        print('Error in reading file, File Path: ', completeName)
                if file_extension == FtypeTXT:
                    try:
                        _RsTxt_(ff, completeName)
                    except PermissionError:
                        errorCount += 1
                        print('Admin Permission Required File Path: ', completeName)
                if file_extension == FtypeDOCX:
                    try:
                        _RsDocx_(ff, completeName)
                    except Exception:
                        errorCount += 1
                        print('Error in reading file, File Path: ', completeName)
                '''
                if file_extension == FtypePDF:
                    try:
                        _rsPDF_(ff, completeName)
                    except Exception:
                        errorCount += 1
                        print('Error in reading file, File Path: ', completeName)
                        '''
                rss.append(completeName)
            i += 1
    else:
        exit()
    x.join()
    logging.info("Master Control RSSF    : Check " + CN + " For result data")
    print('\nSystem scan completed\n><----------------------------------------------><\nSCAN RESULTS')
    # print(rss)
    print('Searched Text:', ff)
    print('Total File Scanned :', len(rss))
    print("Total file found: ", len(rvs))
    print("Total No of unreachable files (ACTION REQUIRED)", errorCount)
    if len(rvs) == 0:
        input("Press enter to close program")
        exit()
    print("Below are the file paths \n", rvs)
    '''
    _Rs_ftp_COPY_()
    _rS_FTP_file_transf(rvs)          #ONLY FOR FTP '''
    print('\n><----------------------------------------------><\ncalling backup')
    _RsBackupWriter(rvs, CN)  # LOCAL BACKUP CREATOR


_rSInternet_Check_()

if _rS_sign():
    print("Other generic information\n><----------------------------------------------><\n1.Backup folders are "
          "ENABLED\n2.FTP options are DISABLED ")
    main()

else:
    print("Author sign missing")

input("Press enter to close program")