You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.3 KiB
53 lines
1.3 KiB
5 years ago
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
"""
|
||
|
|
binaryornot.check
|
||
|
|
-----------------
|
||
|
|
|
||
|
|
Main code for checking if a file is binary or text.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import logging
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
from binaryornot.helpers import get_starting_chunk, is_binary_string
|
||
|
|
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
def is_binary(filename):
|
||
|
|
"""
|
||
|
|
:param filename: File to check.
|
||
|
|
:returns: True if it's a binary file, otherwise False.
|
||
|
|
"""
|
||
|
|
logger.debug('is_binary: %(filename)r', locals())
|
||
|
|
|
||
|
|
# Check if the file extension is in a list of known binary types
|
||
|
|
# binary_extensions = ['.pyc', ]
|
||
|
|
# for ext in binary_extensions:
|
||
|
|
# if filename.endswith(ext):
|
||
|
|
# return True
|
||
|
|
|
||
|
|
# Check if the starting chunk is a binary string
|
||
|
|
chunk = get_starting_chunk(filename)
|
||
|
|
return is_binary_string(chunk)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
parser = argparse.ArgumentParser(description="Check if a "
|
||
|
|
"file passed as argument is "
|
||
|
|
"binary or not")
|
||
|
|
|
||
|
|
parser.add_argument("filename", help="File name to check for. If "
|
||
|
|
"the file is not in the same folder, "
|
||
|
|
"include full path")
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
print(is_binary(**vars(args)))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|