Add Working with File section: Reading, Writing, Updating & Pickling data into/ from a file.

Author: DevOpsThinh

dreariness.txt

@@ -0,0 +1,9 @@
+She love me!
+Yeah
+	Yeah
+		Yeah
+She love me!
+Yeah
+	Yeah
+		Yeah
+(by Truong Thinh)

fundamentals/custom_functions.py

@@ -1,9 +1,105 @@
 # Author: Nguyen Truong Thinh
 # Contact me: [email protected] || +84393280504
 from math import sqrt
+import pickle, os
 
 
 # Custom functions
+def file_with_pickle_data(file_name):
+    """
+    Pickling data for more efficient to use a machine-readable binary file.
+    :param file_name: Name of the file
+    :return: None
+    """
+    if not os.path.isfile(file_name):
+        data = [0, 1]
+        data[0] = input('Enter topic: ')
+        data[1] = input('Enter series: ')
+
+        file = open(file_name, 'wb')
+        pickle.dump(data, file)
+        file.close()
+    else:
+        file = open(file_name, 'rb')
+        data = pickle.load(file)
+        file.close()
+        print('\nWelcome back to: ', data[0], '-', data[1])
+
+
+def update_a_file_with_block(file_name, mode, new_content, position):
+    """
+    Updating file strings
+    :param file_name: Name of the file
+    :param mode: File mode
+    :param new_content: The new content will be updated
+    :param position: The current file position
+    :return: None
+    """
+    with open(file_name, mode) as file:
+        text = file.read()
+        print('\nString: ', text)
+        print('\nPosition in file now: ', file.tell())
+        file.seek(position)
+        file.write(new_content)
+        file.seek(0)
+        text = file.read()
+        print('\nString: ', text)
+        print('\nPosition in file now: ', file.tell())
+
+
+def write_a_file_with_block(file_name, mode, content):
+    """
+    Writing files: Adds content to the file & display the file's current status
+    in the "with" block.
+    :param file_name: Name of the file
+    :param mode: File mode
+    :param content: The content will be written
+    :return: None
+    """
+    with open(file_name, mode) as file:
+        file.write(content)
+        print('\nFile Now Closed?: ', file.closed)
+    print('File Now Closed?: ', file.closed)
+
+
+def re_write_a_file(file_name):
+    """
+    Re-write a file to append a citation
+    :param file_name: Name of the file
+    :return: None
+    """
+    file = open(file_name, 'a')
+    file.write('(by Truong Thinh)')
+    file.close()
+
+
+def read_a_file(file_name, mode):
+    """
+    Reading files: Read the entire contents
+    :param file_name: Name of the file
+    :param mode: File mode
+    :return: None
+    """
+    file = open(file_name, mode)
+    for line in file:
+        # Do something here.
+        print(line, end='')
+    file.close()
+
+
+def write_a_file(file_name, mode, content):
+    """
+    Writing files: Adds content to the file
+    :param file_name: Name of the file
+    :param mode: File mode
+    :param content: The content will be written
+    :return: None
+    """
+    file = open(file_name, mode)
+    file.write(content)
+    file.close()
+
+
 def get_data_from_list(stocks):
     """
     Get a list of:

main.py

@@ -76,11 +76,23 @@
     friends_stock_list = [info_dict_dev, info_dict_tester, info_dict_po, info_dict_boss]
     data = get_data_from_list(friends_stock_list)
     print(data)
+    # File operations
+    file_name = 'dreariness.txt'
+    content = 'She love me!\n'
+    content += 'Yeah\n'
+    content += '\tYeah\n'
+    content += '\t\tYeah\n'
+    write_a_file(file_name, 'w', content)
+    read_a_file(file_name, 'r')
+    update_a_file_with_block(file_name, 'r+', content, 35)
+    re_write_a_file(file_name)
+    file_with_pickle_data('pickle.dat')
+
     # Objects
     p_x = Point(1, 0)
     p_y = Point(5, 3)
-    # draw(p_x.x, p_x.y)
-    # draw(p_y.x, p_y.y)
+    draw(p_x.x, p_x.y)
+    draw(p_y.x, p_y.y)
     print(f'\nDistance between {p_x} & {p_y} is {p_x.distance(p_y)}')
     del p_x
     del p_y