sourav-das/stem-separator
3
1�
2 Ԏ�i� � �b � S SK r S SKrS SKrS SKrS SKrS SKrS SKJr \" S5 rSr Sr3S\ 4S jr S\ S\4S jr
S\ S4\ S \S\4S jrS\ S
\ S \S\4S jrS\ S\S- 4S jrS\ S\4S jrS\ S\S\4S jrS\ S\S- 4S jrS\ S\4S jrS\ S\\ \ 4 4S jrS\ S5\ S\S- 4S jrS\ 4S jrS rg)� N)�Pathz
/tmp/stem-sepi i, �returnc �� � [ R " 5 R S S n [ U - nUR SSS9 US- R [
[ R " 5 5 5 U $ )N� T��parents�exist_ok�.created)�uuid�uuid4�hex�BASE_DIR�mkdir�6write_text�str�time��job_id�job_dirs �backend\file_manager.py�7create_jobr sW � �
�Z�Z�\�
�
�c�r�
"�F��� �G� �M�M�$��M�.� �z��%�%�c�$�)�)�+�&6�7� �M� r c � � [ U - $ �N)r �r s r � get_job_dirr s
� � �f� � r �filename�contentc � � [ U 5 n[ U5 R =( d SnUSU 3- nUR U5 U$ )N�.wav�input)r r �suffix� write_bytes)r r r r �ext�8input_paths r � save_uploadr& sD � ��&�!�G�9�x�.�10 �11 �12)�6�C��U�3�%�=�(�J����7�#� �r �source_filenamec � � [ XU5 $ r )r& )r r' r s r �save_imported_audior) # s � � �v�� 8�8r c �� � [ U 5 nUR 5 ( d g UR 5 H> nUR 5 ( d M UR R S5 ( d M< Us $ g )Nr! )r �exists�iterdir�is_file�name�13startswith)r r �files r �get_input_filer1 ' sV � ��&�!�G� �>�>� � �����!�� �<�<�>�>�d�i�i� 2� 2�7� ;� ;��K� "� r c � � [ U 5 S- $ )Nz
metadata.json)r r s r �get_metadata_pathr3 2 s � � �v� � � 0�0r �metadatac �b � [ U 5 nUR [ R " USS9SS9 U$ )N� )�indent�utf-8��encoding)r3 r �json�dumps)r r4 �paths r �save_job_metadatar> 6 s, � � �V� $�D��O�O�D�J�J�x��2�W�O�E� �Kr c � � [ U 5 nUR 5 ( d g [ R " UR SS95 $ )Nr8 r9 )r3 r+ r; �loads� read_text)r r= s r �load_job_metadatarB < s4 � � �V� $�D� �;�;�=�=�� �:�:�d�n�n�g�n�6� 7�7r c �B � [ U 5 S- nUR SSS9 U$ )N�stemsTr )r r )r �outs r �get_output_dirrF C s& � �14�f�15 ��16'�C��I�I�d�T�I�*� �Jr c � � 0 n1 Skn[ U 5 [ U 5 4 H� nUR 5 ( d M UR 5 H� nUR 5 ( d M UR17 R
5 U; d M: UR R S5 ( a M\ UR U; d Mn UR XR ' M� M� U$ )z>Find all stem audio files, checking stems subdir and job root.> �.aac�.mp3r �.flacr! )18rF r r+ r, r- r" �lowerr. r/ �stem)r �result�19audio_exts� check_dir�fs r �list_stem_filesrQ I s� � �
�F�2�J�$�V�,�k�&�.A�B� �� � �!�!� ��"�"�$�A��y�y�{�{�q�x�x�~�~� /�:� =�a�f�f�FW�FW�X_�F`�F`�ef�ek�ek�sy�ey�!"����v�v�� %� C� �Mr c � � [ U 5 U- nUR 5 ( a U$ [ U 5 U- nUR 5 ( a U$ g r )rF r+ r )r r � stem_path� root_paths r �
get_file_pathrU V sM � � �v�&��1�I���������F�#�h�.�I�������� r c �p � [ U 5 nUR 5 ( a [ R " USS9 g g )NT��
ignore_errors)r r+ �shutil�rmtreer s r �20delete_jobr[ a s, � ��&�!�G��~�~����
�
�g�T�2� r c � � # � [ R " [ 5 I Sh v�N [ R 5 ( d M? [21 R22 " 5 n [ R
5 H� nUR 5 ( d M US- nUR 5 ( aL [ UR 5 R 5 5 nX-23 [ :� a [ R " USS9 M~ M� XR 5 R -24 [ :� d M� [ R " USS9 M� GM) GN ! [ a Nf = f7f)z<Background task: delete job dirs older than MAX_AGE_SECONDS.TNr25 rW )�asyncio�sleep�CLEANUP_INTERVALr r+ r r, �is_dir�floatrA �strip�MAX_AGE_SECONDSrY rZ �stat�st_mtime� Exception)�now�entry�marker�createds r � cleanup_looprk g s� � � �26��m�m� ,�-�-�-� ��?�?�$�$���)�)�+�C�!�)�)�+�� �|�|�~�~� � ��+���=�=�?�?� #�F�$4�$4�$6�$<�$<�$>� ?�G��}��6� �
�
�e�4�@� 7� �Z�Z�\� 2� 2�2�_�D� �
�
�e�4�@� ,�
�-��" � � � �sE � E �D-�E �D0 � E �C D0 �D0 �*E �027D=�:E �<D=�=E )�osrY r r r] r; �pathlibr r rc r_ r r r �bytesr& r) r1 r3 �dictr> rB rF rQ rU r[ rk � r r �<module>rq sH �� �
� � � � � � �� ����� ��C� � �� �� ��� �s� �U� �t� �9�� 9�c� 9�E� 9�d� 9��3� �4�$�;� �1�c� 1�d� 1��c� �T� �d� � 8�c� 8�d�T�k� 8��3� �4� � 28�C� 29�D��c��N� 30��#� �� ���� �3�s� 3� r