slightly improved readstatic

This commit is contained in:
Kevin Froman 2019-09-21 21:45:36 -05:00
parent 021954c86b
commit 921bf36a3f
1 changed files with 25 additions and 3 deletions

View File

@ -1,7 +1,29 @@
"""
Onionr - Private P2P Communication
module to get static directory and read static data files
"""
"""
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
"""
from typing import Union
import os
def get_static_dir():
def get_static_dir()->str:
return os.path.dirname(os.path.realpath(__file__)) + '/../static-data/'
def read_static(file:str, ret_bin:bool=False)->str:
def read_static(file:str, ret_bin:bool=False)->Union[str, bytes]:
static_file = get_static_dir() + file
if ret_bin:
@ -10,4 +32,4 @@ def read_static(file:str, ret_bin:bool=False)->str:
mode = 'r'
with open(static_file, mode) as f:
data = f.read()
return data
return data