"""QGIS Plugin Implementation."""

def __init__(self, iface):

"""Constructor.

:param iface: An interface instance that will be passed to this class

which provides the hook by which you can manipulate the QGIS

application at run time.

:type iface: QgsInterface

"""

# Save reference to the QGIS interface

self.iface = iface

# initialize plugin directory

self.plugin_dir = os.path.dirname(__file__)

# initialize locale

locale = QSettings().value('locale/userLocale')[0:2]

locale_path = os.path.join(

self.plugin_dir,

'i18n',

'finddata_{}.qm'.format(locale))

if os.path.exists(locale_path):

self.translator = QTranslator()

self.translator.load(locale_path)

QCoreApplication.installTranslator(self.translator)

# Declare instance attributes

self.actions = []

self.menu = self.tr(u'&finddata')

self.toolbar = self.iface.addToolBar(u'finddata')

self.toolbar.setObjectName(u'finddata')

# Check if plugin was started the first time in current QGIS session

# Must be set in initGui() to survive plugin reloads

self.first_start = None

self.search_folder = ""

self.file_log = "" #Log file

self.file_csv = "" # output csv file

# noinspection PyMethodMayBeStatic

def tr(self, message):

"""Get the translation for a string using Qt translation API.

We implement this ourselves since we do not inherit QObject.

:param message: String for translation.

:type message: str, QString

:returns: Translated version of message.

:rtype: QString

"""

# noinspection PyTypeChecker,PyArgumentList,PyCallByClass

return QCoreApplication.translate('finddata', message)

def add_action(

self,

icon_path,

text,

callback,

enabled_flag=True,

add_to_menu=True,

add_to_toolbar=True,

status_tip=None,

whats_this=None,

parent=None):

"""Add a toolbar icon to the toolbar.

:param icon_path: Path to the icon for this action. Can be a resource

path (e.g. ':/plugins/foo/bar.png') or a normal file system path.

:type icon_path: str

:param text: Text that should be shown in menu items for this action.

:type text: str

:param callback: Function to be called when the action is triggered.

:type callback: function

:param enabled_flag: A flag indicating if the action should be enabled

by default. Defaults to True.

:type enabled_flag: bool

:param add_to_menu: Flag indicating whether the action should also

be added to the menu. Defaults to True.

:type add_to_menu: bool

:param add_to_toolbar: Flag indicating whether the action should also

be added to the toolbar. Defaults to True.

:type add_to_toolbar: bool

:param status_tip: Optional text to show in a popup when mouse pointer

hovers over the action.

:type status_tip: str

:param parent: Parent widget for the new action. Defaults None.

:type parent: QWidget

:param whats_this: Optional text to show in the status bar when the

mouse pointer hovers over the action.

:returns: The action that was created. Note that the action is also

added to self.actions list.

:rtype: QAction

"""

icon = QIcon(icon_path)

action = QAction(icon, text, parent)

action.triggered.connect(callback)

action.setEnabled(enabled_flag)

if status_tip is not None:

action.setStatusTip(status_tip)

if whats_this is not None:

action.setWhatsThis(whats_this)

if add_to_toolbar:

# Adds plugin icon to Plugins toolbar

self.iface.addToolBarIcon(action)

if add_to_menu:

self.iface.addPluginToMenu(

self.menu,

action)

self.actions.append(action)

return action

def search_vector_data(self):

data_vector = ['shp', 'mif', 'mid']

search_folder = self.search_folder

# add to log

self.dlg.lvLog.addItem("Starting at: ")

self.dlg.lvLog.addItem(search_folder)

numRows = self.dlg.tableWidget.rowCount()

# Create a empty row at bottom of table

#numRows = self.dlg.tableWidget.rowCount()

#self.dlg.tableWidget.insertRow(numRows)

#self.dlg.tableWidget.show()

# Add text to the row

try:

for root, subdirs, files in os.walk(search_folder):

for file in os.listdir(root):

file_path = str(os.path.join(root, file))

if os.path.isfile(file_path):

numRows = self.dlg.tableWidget.rowCount()

self.dlg.tableWidget.insertRow(numRows)

# .lower() - под линуксом есть разница!!!

# ext = '.'.join(file.split('.')[1:]).lower()

file_name = file # .lower()

print(file_name)

# file_ext = self.get_extension(file)

ext = '.'.join(file.split('.')[1:]).lower()

# 'shx','shp'

self.dlg.lvLog.addItem(ext)

self.dlg.tableWidget.setItem(numRows, 0, QtWidgets.QTableWidgetItem(file_name))

self.dlg.tableWidget.setItem(numRows, 1, QtWidgets.QTableWidgetItem(ext))

self.dlg.tableWidget.setItem(numRows, 2, QtWidgets.QTableWidgetItem("size"))

self.dlg.tableWidget.setItem(numRows, 3, QtWidgets.QTableWidgetItem("file_name"))

self.dlg.tableWidget.setItem(numRows, 4, QtWidgets.QTableWidgetItem("cdata"))

self.dlg.tableWidget.setItem(numRows, 5, QtWidgets.QTableWidgetItem("mdata"))

self.dlg.tableWidget.setItem(numRows, 6, QtWidgets.QTableWidgetItem("ladata"))

self.dlg.tableWidget.setItem(numRows, 7, QtWidgets.QTableWidgetItem(file_path))

self.dlg.tableWidget.setItem(numRows, 8, QtWidgets.QTableWidgetItem("npath"))

self.dlg.tableWidget.setHorizontalHeaderItem(2, QtWidgets.QTableWidgetItem())

# if os.path.isfile(file_path) : # and file_name.startswith('info.doc') #ext == "csv":

# self.dlg.lvLog.addItem(file_path)

#listdir.append(file_path)

except Exception as e:

ss = "Exception occurred search_vector_data" + str(e)

print(ss)

self.dlg.lvLog.addItem(ss)

QMessageBox.information(None, "Info!", "Done!")

pass

def get_extension(filename=''):

basename = os.path.basename(filename) # os independent

ext = '.'.join(basename.split('.')[1:])

return str('' + ext if ext else '').lower()

def select_root_folder(self):

foldername = QFileDialog.getExistingDirectory(self.dlg, "Select folder ","",)

#print("select_root_folder(self)")

if os.path.exists(foldername):

self.dlg.edtFolder.setText(foldername)

self.search_folder = foldername

self.search_vector_data()

def search_spatial_data(self):

root_folder = self.dlg.edtFolder.displayText()

if root_folder == '':

QMessageBox.information(None, "Warning!", "No root folder selected. Please select a folder.")

return

else:

QMessageBox.information(None, "Warning!", root_folder)

def initGui(self):

"""Create the menu entries and toolbar icons inside the QGIS GUI."""

icon_path = ':/plugins/finddata/icon.png'

self.add_action(

icon_path,

text=self.tr(u'finddata'),

callback=self.run,

parent=self.iface.mainWindow())

# will be set False in run()

self.first_start = True

def unload(self):

"""Removes the plugin menu item and icon from QGIS GUI."""

for action in self.actions:

self.iface.removePluginMenu(

self.tr(u'&finddata'),

action)

self.iface.removeToolBarIcon(action)

def run(self):

"""Run method that performs all the real work"""

# Create the dialog with elements (after translation) and keep reference

# Only create GUI ONCE in callback, so that it will only load when the plugin is started

if self.first_start == True:

self.first_start = False

self.dlg = finddataDialog()

# self.dlg.tableWidget.columnCount()

# Set column headers

self.dlg.tableWidget.setHorizontalHeaderItem(0, QtWidgets.QTableWidgetItem("filename"))

self.dlg.tableWidget.setHorizontalHeaderItem(1, QtWidgets.QTableWidgetItem("ext"))

self.dlg.tableWidget.setHorizontalHeaderItem(2, QtWidgets.QTableWidgetItem("size"))

self.dlg.tableWidget.setHorizontalHeaderItem(3, QtWidgets.QTableWidgetItem("hsize"))

self.dlg.tableWidget.setHorizontalHeaderItem(4, QtWidgets.QTableWidgetItem("cdata"))

self.dlg.tableWidget.setHorizontalHeaderItem(5, QtWidgets.QTableWidgetItem("mdata"))

self.dlg.tableWidget.setHorizontalHeaderItem(6, QtWidgets.QTableWidgetItem("ladata"))

self.dlg.tableWidget.setHorizontalHeaderItem(7, QtWidgets.QTableWidgetItem("fpath"))

self.dlg.tableWidget.setHorizontalHeaderItem(8, QtWidgets.QTableWidgetItem("npath"))

self.dlg.edtFolder.clear()

#self.dlg

# signals

self.dlg.btnSelectFolder.pressed.connect(self.select_root_folder)

self.dlg.btnApply.pressed.connect(self.search_spatial_data)

# show the dialog

self.dlg.show()

# Run the dialog event loop

result = self.dlg.exec_()

# See if OK was pressed

if result:

# Do something useful here - delete the line containing pass and

# substitute with your code.