pMi1 z d Z ddlZddlmZmZmZmZmZmZm Z m
Z
m Z ddl m
Z
mZ ddlmZ ddlmZ G d de Zy) z,Aggregator tab for combining markdown files. N) QWidget QVBoxLayout QHBoxLayoutQLabel QLineEdit QPushButton QProgressBar QFileDialog QMessageBox)QtQCoreApplication)
Aggregator)ObsidianToolExceptionc F e Zd ZdZ fdZd Zd Zd Zd Zde fdZ
xZ S )
AggregatorTabz,Tab for aggregating multiple markdown files.c ` t | t | _ | j y)z Initialize the aggregator tab.N)super__init__r
aggregatorinit_ui)self __class__s ?/mnt/ssd/data/python-lab/ObsidianTool/src/gui/aggregator_tab.pyr zAggregatorTab.__init__ s
$, c 4 t | }|j d |j dddd |j t d t }t
| _ | j j d | j j d |j | j t d | _ | j j d | j j j | j |j | j |j! | |j t d t }t
| _ | j" j d | j" j d |j | j" t d | _ | j$ j d | j$ j j | j& |j | j$ |j! | |j) d t d
| _ | j* j- d | j* j j | j. |j | j* t1 | _ | j2 j5 d |j | j2 t d
| _ | j6 j9 t: j< j> | j6 jA d |j | j6 |jC y)z Initialize the user interface. zSource Directory:Tz*Select a directory containing .md files...z Browse...d zOutput File:z Select output file location...zAggregate Files( r Readyz color: #666; font-style: italic;N)"r
setSpacingsetContentsMargins addWidgetr r r source_edit setReadOnlysetPlaceholderTextr
source_btnsetMaximumWidthclickedconnect
browse_source addLayout output_edit
output_btn
browse_output
addSpacing
aggregate_btnsetMinimumHeightaggregate_filesr progress_barsetValue status_label setAlignmentr
AlignmentFlag AlignCenter
setStyleSheet
addStretch)r layout
source_layout
output_layouts r r zAggregatorTab.init_ui s T"" !!"b"b1 :;<#
$; $$T* ++,XY 0 01%k2 '', ''(:(:; 0 ' 567#
$; $$T* ++,LM 0 01%k2 '', ''(:(:; 0 ' " )): ; ++B/ ""**4+?+?@++, )N ""1%**+ #7O &&r'7'7'C'CD ''(JK**+ r c t j | dt j j d }|r7| j
j
| | j j
d yy)z'Open dialog to select source directory.zSelect Source Directory~zSource directory selectedN)r
getExistingDirectoryospath
expanduserr$ setTextr6 )r directorys r r+ zAggregatorTab.browse_sourceS s] 44 % GG s #
$ $Y / % %&A B r c t j | dt j j d d \ }}|rM|j d s|dz
}| j j | | j j d yy)z"Open dialog to select output file.zSelect Output Filez~/aggregated.mdz$Markdown Files (*.md);;All Files (*)z.mdzOutput file selectedN) r
getSaveFileNamerB rC rD endswithr- rE r6 )r file_path_s r r/ zAggregatorTab.browse_output_ s} "22 GG 0 1 2
1 %%e,U" $ $Y / % %&< = r c & j j } j j }|st j dd y|st j dd y j d j j d j j d fd} j j ||| }t j d d
|d d |d
j j d j d y# t $ rc}t j ddt | j j d j j d Y d}~yd}~wt $ rc}t j ddt | j j d j j d Y d}~d}~ww xY w# j d w xY w)z"Perform the aggregation operation.zInput Requiredz!Please select a source directory.Nz Please select an output file.Fr zAggregating files...c j j | j j | t j
y )N)r4 r5 r6 rE r
processEvents)
percentagemessager s r update_progressz6AggregatorTab.aggregate_files..update_progress s7 !!**:6!!))'2 ..0r Successz(Aggregation complete!
Files processed: files_processedz
Output file: output_filez"Aggregation completed successfullyErrorzAggregation failed:
zAggregation failedzUnexpected Errorz An unexpected error occurred:
T)r$ textr- r warningset_ui_enabledr4 r5 r6 rE r aggregate informationr criticalstr Exception)r
source_dirrT rQ resultes ` r r3 z AggregatorTab.aggregate_fileso s %%**,
&&++- 3
/
E" ""1% !!"89- &
1 __.. F
# #$$*+<$=#> ? &} 5 68
% %&J K,
%) % * )#a&2
% %&: ; & &q ) ) * "3CF8<
% %&: ; & &q ) ) *
%s? 2A D$ $ G:-AF G= G:AG50G= 5G::G= =Henabledc | j j | | j j | | j j | y)z/Enable or disable UI elements during operation.N)r'
setEnabledr. r1 )r ra s r rX z AggregatorTab.set_ui_enabled s: ""7+ ""7+ %%g.r ) __name__
__module__ __qualname____doc__r r r+ r/ r3 boolrX
__classcell__)r s @r r r
s/ 6 ; z
C> H&T/d /r r )rg rB PyQt6.QtWidgetsr r r r r r r r
r PyQt6.QtCorer r
core.aggregatorr core.exceptionsr r r r ro s3 2 . & 1p/G p/r