forked from adamm/git-importer
Extend documention and use some more pythonier loops
This commit is contained in:
parent
05a5e6aea7
commit
a1ead29734
@ -1,5 +1,8 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
from hashlib import md5
|
||||
from typing import Optional
|
||||
|
||||
from lib.db import DB
|
||||
from lib.request import Request
|
||||
@ -204,18 +207,24 @@ class DBRevision:
|
||||
self._files.sort(key=lambda x: x["name"])
|
||||
return self._files
|
||||
|
||||
def calc_delta(self, db: DB, current_rev=None):
|
||||
"""Calculate the list of files to download and to delete"""
|
||||
old_files = dict()
|
||||
def calc_delta(self, db: DB, current_rev: Optional[DBRevision]):
|
||||
"""Calculate the list of files to download and to delete.
|
||||
Param current_rev is the revision that's currently checked out.
|
||||
If it's None, the repository is empty.
|
||||
"""
|
||||
to_download = []
|
||||
to_delete = []
|
||||
if current_rev:
|
||||
for entry in current_rev.files_list(db):
|
||||
old_files[entry["name"]] = f"{entry['md5']}-{entry['size']}"
|
||||
old_files = {
|
||||
e["name"]: f"{entry['md5']}-{entry['size']}"
|
||||
for e in current_rev.files_list(db)
|
||||
}
|
||||
else:
|
||||
old_files = dict()
|
||||
for entry in self.files_list(db):
|
||||
if old_files.get(entry["name"], "") != f"{entry['md5']}-{entry['size']}":
|
||||
if old_files.get(entry["name"]) != f"{entry['md5']}-{entry['size']}":
|
||||
logging.debug(f"Download {entry['name']}")
|
||||
to_download.append(entry["name"])
|
||||
to_download.append((entry["name"], entry["md5"]))
|
||||
old_files.pop(entry["name"], None)
|
||||
for entry in old_files.keys():
|
||||
logging.debug(f"Delete {entry}")
|
||||
|
Loading…
Reference in New Issue
Block a user