Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-01-26 19:09:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Remove successful model downloads (#1710)

Browse Source 
* feat: Remove successful model downloads

* PR comments

---------

Co-authored-by: Oto Ciulis <oto.ciulis@gmail.com>
This commit is contained in:
oto-ciulis-tt
2024-11-27 07:40:43 -08:00
committed by GitHub
parent 1ac6d6529f
commit 2d022e4e49
3 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/components/sidebar/tabs/modelLibrary/ElectronDownloadItems.vue
View File

@@ -1,10 +1,10 @@
<template> <template>
<div class="mx-6 mb-4" v-if="downloads.length > 0"> <div class="mx-6 mb-4" v-if="inProgressDownloads.length > 0">
<div class="text-lg my-4"> <div class="text-lg my-4">
{{ $t('electronFileDownload.inProgress') }} {{ $t('electronFileDownload.inProgress') }}
</div> </div>
<template v-for="download in downloads" :key="download.url"> <template v-for="download in inProgressDownloads" :key="download.url">
<DownloadItem :download="download" /> <DownloadItem :download="download" />
</template> </template>
</div> </div>
@@ -16,5 +16,5 @@ import { useElectronDownloadStore } from '@/stores/electronDownloadStore'
import { storeToRefs } from 'pinia' import { storeToRefs } from 'pinia'
const electronDownloadStore = useElectronDownloadStore() const electronDownloadStore = useElectronDownloadStore()
const { downloads } = storeToRefs(electronDownloadStore) const { inProgressDownloads } = storeToRefs(electronDownloadStore)
</script> </script>

4
src/hooks/sidebarTabs/modelLibrarySidebarTab.ts
View File

@@ -18,8 +18,8 @@ export const useModelLibrarySidebarTab = (): SidebarTabExtension => {
iconBadge: () => { iconBadge: () => {
if (isElectron()) { if (isElectron()) {
const electronDownloadStore = useElectronDownloadStore() const electronDownloadStore = useElectronDownloadStore()
if (electronDownloadStore.downloads.length > 0) { if (electronDownloadStore.inProgressDownloads.length > 0) {
return electronDownloadStore.downloads.length.toString() return electronDownloadStore.inProgressDownloads.length.toString()
} }
} }

13
src/stores/electronDownloadStore.ts
View File

@@ -1,8 +1,8 @@
import { ref } from 'vue' import { computed, ref } from 'vue'
import { defineStore } from 'pinia' import { defineStore } from 'pinia'
import { isElectron, electronAPI } from '@/utils/envUtil' import { isElectron, electronAPI } from '@/utils/envUtil'
import type { import {
DownloadState, type DownloadState,
DownloadStatus DownloadStatus
} from '@comfyorg/comfyui-electron-types' } from '@comfyorg/comfyui-electron-types'
@@ -69,6 +69,11 @@ export const useElectronDownloadStore = defineStore('downloads', () => {
resume, resume,
cancel, cancel,
findByUrl, findByUrl,
initialize initialize,
inProgressDownloads: computed(() =>
downloads.value.filter(
({ status }) => status !== DownloadStatus.COMPLETED
)
)
} }
}) })