Niv Sardi commited on
Commit
4b890a6
1 Parent(s): e34b2c2

Signed-off-by: Niv Sardi <xaiki@debian-BULLSEYE-live-builder-AMD64>

Files changed (2) hide show
  1. python/entity.py +2 -2
  2. python/web.py +4 -5
python/entity.py CHANGED
@@ -12,8 +12,8 @@ class Entity(NamedTuple):
12
  name: str
13
  id: int = 0
14
  bco: str = "debug"
15
- url: str = None
16
- logo: str = None
17
 
18
  def __repr__(self):
19
  return f"""
 
12
  name: str
13
  id: int = 0
14
  bco: str = "debug"
15
+ url: str = ''
16
+ logo: str = ''
17
 
18
  def __repr__(self):
19
  return f"""
python/web.py CHANGED
@@ -27,11 +27,8 @@ def get_cert(e: Entity):
27
  f.write(str(err))
28
  return fn
29
 
30
- def get_img_logo(src: string):
31
- ext = src.split('.')[-1].split('/')[-1]
32
  res = requests.get(src, stream=True)
33
-
34
- fn = f"{defaults.DATA_PATH}/logos/{e.bco}.{i}.{ext}"
35
  with open(fn, "wb") as f:
36
  shutil.copyfileobj(res.raw, f)
37
  return fn
@@ -47,7 +44,9 @@ def get_logos(e: Entity, page):
47
  for l in logos:
48
  if 'src' in l.attrs:
49
  src = l.attrs['src']
 
50
  if not src.startswith('http'): src = e.url + src
51
- lfn.append(get_img_logo(src))
 
52
  i+=1
53
  return lfn
 
27
  f.write(str(err))
28
  return fn
29
 
30
+ def get_img_logo(src: str, fn):
 
31
  res = requests.get(src, stream=True)
 
 
32
  with open(fn, "wb") as f:
33
  shutil.copyfileobj(res.raw, f)
34
  return fn
 
44
  for l in logos:
45
  if 'src' in l.attrs:
46
  src = l.attrs['src']
47
+ ext = src.split('.')[-1].split('/')[-1]
48
  if not src.startswith('http'): src = e.url + src
49
+ fn = f"{defaults.DATA_PATH}/logos/{e.bco}.{i}.{ext}"
50
+ lfn.append(get_img_logo(src, fn))
51
  i+=1
52
  return lfn