Update data_get_store

lee-coder · web-flow · commit 8f529ba98f26 · 2016-10-11T22:22:49.000+08:00
diff --git a/data_get_store b/data_get_store
@@ -56,11 +56,29 @@ def CompareTwoElements(x,y,order=0):    # 0 means left smaller than right, 1 mea
 
 def JudgeOrderForArray(array,order=0):  # order 0 means small to large, 1 means large to small
     i = 0
+    if(len(array)==0):
+        return 1
     last_index = len(array)-1
     while(i<last_index and CompareTwoElements(array[i],array[i+1],order)):
         i = i+1
     return i==last_index
 
+'''
+given a array, use following function to judge it is a ordered array or not
+if it is ordered, gives its order:
+0 represent small to large, 1 means large to small
+if it it not ordered, the order is None
+'''
+def JudgeAndGiveOrderForArray(array):
+    i=0
+    while((i<len(array)-1) and array[i]==array[i+1]):
+        i = i+1
+    remain_arr=array[i:]
+    order = (0 if (i==(len(array)-1) or len(array)==0) else (array[i]>array[i+1]))
+    is_order = JudgeOrderForArray(remain_arr,order)
+    order_val = int(order) if is_order else None
+    return bool(is_order),order_val
+
 def OrderedDictReorder(ord_dict,index="key",rev=False):
     from collections import OrderedDict
     return OrderedDict(sorted(ord_dict.items(), key=lambda t:t[0] if index=="key" else t[1],reverse=rev))
@@ -72,6 +90,26 @@ def WriteNewDictDataToFile(ord_dict,txt_str=""):
         datafile.write("%d\t%.4f\n" % (index, ord_dict[index]))
     datafile.close()
 
+'''
+before call this function, use above functions to reorder the new_data
+to a new txt file, and the old txt file has been reordred,
+so this function only need to judge the order and decide to merge or add
+the new txt data to old txt from which position(start or end)
+'''
+def AddNewDataToFile(old_data_txt="",add_data_txt="",new_data_txt=""):
+    from collections import OrderedDict
+    old_key_list = GetIndexToList(GetDataFromTxt(old_data_txt))
+    add_key_list = GetIndexToList(GetDataFromTxt(add_data_txt))
+    old_is_order,old_order_val = JudgeAndGiveOrderForArray(old_key_list)
+    add_is_order,add_order_val = JudgeAndGiveOrderForArray(add_key_list)
+    if(old_is_order==False):
+        return ("Please check the data in %s, it is not ordered!" %s (old_data_txt))
+    if(add_is_order==False):
+        return ("Please check the data in %s, it is not ordered!" %s (add_data_txt))
+    if(old_order_val!=1):
+        return ("I hope the data in %s order is larger to small for date key!" %s (old_data_txt))
+    if(add_order_val!=1):
+        return ("I hope the data in %s order is larger to small for date key!" %s (add_data_txt))
     
 if __name__ == "__main__":   
     raw_data = GetDataFromTxt("xxx.txt")
@@ -111,3 +149,6 @@ if __name__ == "__main__":
 
 #print(CompareTwoElements(1,1,0))
 #print(JudgeOrderForArray([4,3,3,2],1))
+
+print(JudgeAndGiveOrderForArray([4,3,2,1,1,1]))
+print(JudgeAndGiveOrderForArray([1]))
